requirement timm 0.6.12 dataset train CG Real test CG Real train and test python train.py python test.py