libffcv / ffcv-imagenet

Train ImageNet *fast* in 500 lines of code with FFCV
Apache License 2.0
136 stars 34 forks source link

ValueError(“total size of new array must be changed”) #12

Open cindy-cheng0214 opened 2 years ago

cindy-cheng0214 commented 2 years ago

Hi, I’m trying to train model by the guide line. But I got the ValueError and SystemError when I try to load the following codes:

python train_imagenet.py --config-file rn50_configs/<your config file>.yaml \ --data.train_dataset=/path/to/train/dataset.ffcv \ --data.val_dataset=/path/to/val/dataset.ffcv \ --data.num_workers=12 --data.in_memory=1 \ --logging.folder=/your/path/here

截圖 2022-05-23 下午1 27 25 截圖 2022-05-23 下午1 27 32

How can I solve this? Thanks in advance for your replies.