search

vt-le / astnet

This is an official implementation for "Attention-based Residual Autoencoder for Video Anomaly Detection".
https://vt-le.github.io/astnet/
MIT License
102 stars 15 forks source link

Training error #23

Open Akahori000 opened 6 months ago

Akahori000 commented 6 months ago

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)

image

vt-le commented 5 months ago

Hi, Thanks for your interest!

Please, check this line