Training error

[Akahori000]

Hello, Thank you for the code! I have tried training with the ped2 dataset, but I got this error with the default pred2_wresnet.yaml.

python train.py \
    --cfg config/ped2_wresmet.yaml

RuntimeError: Given groups=1, weight of size [2048, 16384, 1, 1], expected input[2, 8192, 28, 36] to have 16384 channels, but got 8192 channels instead

Could you provide me with ped2_wresnet.yaml that works for the training of UCSped2? (http://www.svcl.ucsd.edu/projects/anomaly/dataset.html)

[vt-le]

Hi, Thanks for your interest!

Please, check this line