zlyx525

This is because your pictures are not "*.jpg" format, your can set the option according to your own format, like this: `python main.py --dataset=flowares --input_fname_pattern="*.png" --train `

In train.py, the grid_size is reset. So, when img_size = 224, the grid_size is 224//116=14, and the patch_size=224//14//16 = 1. ` if args.vit_name.find('R50') != -1: ` `config_vit.patches.grid = (int(args.img_size /...