I used my own small dataset to extract I3D features. Then use main.py to train. Overfitting happened in training set. Do you have any suggestions? 