AOGNet
AOGNet copied to clipboard
error when training by cifar10
Hi,I try to rerun this code to test this model's performance by using the 'python3.6 main.py --cfg cfgs/cifar10/aognet_cifar10_ps_4_bottleneck_1M.yaml --gpus 1,2'.At first everything seemed to be going smoothly,however,when it comes to epoch 280,it is stoped by an error:
Traceback (most recent call last):
File "main.py", line 133, in
sorry, I didn't have this problem before. It seems like an error to save checkpoint to file. Anyway, the code is based on old version of mxnet. The results cannot match the results reported in the paper. We'll release our new pytorch code very soon. Stay tuned.