bugs when eval

[cuge1995]

2021-06-26 22:30:03,174:[P:3190309]:Rank[0/5] => switch to eval mode
2021-06-26 22:30:03,175:[P:3190312]:Rank[3/5] => Epoch[0]: train end, duration: 8942.97s
2021-06-26 22:30:03,176:[P:3190312]:Rank[3/5] => Epoch[0]: validate start
2021-06-26 22:30:03,176:[P:3190312]:Rank[3/5] => switch to eval mode
Traceback (most recent call last):
  File "tools/train.py", line 211, in <module>
    main()
  File "tools/train.py", line 153, in main
    args.distributed
  File "/python3.6/site-packages/torch/autograd/grad_mode.py", line 26, in decorate_context
    return func(*args, **kwargs)
  File "/CvT/tools/../lib/core/function.py", line 133, in test
    loss = criterion(outputs, y)