train ucsd

[DengMiao55]

Hi，thanks for your work。When I train it ,I get this error。

Traceback (most recent call last):
File "train.py", line 201, in
main()
File "train.py", line 192, in main
train_ucsdped2()
File "train.py", line 128, in train_ucsdped2
x_r, z, z_dist = model(x)
File "/home/dl/VSST/dm/novelty-detection-master/models/base.py", line 33, in call
return super(BaseModule, self).call(*args, **kwargs)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/dl/VSST/dm/novelty-detection-master/models/LSA_ucsd.py", line 189, in forward
z = self.encoder(h)
File "/home/dl/VSST/dm/novelty-detection-master/models/base.py", line 33, in call
return super(BaseModule, self).call(*args, **kwargs)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/dl/VSST/dm/novelty-detection-master/models/LSA_ucsd.py", line 62, in forward
h = self.conv(h)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/container.py", line 91, in forward
input = module(input)
File "/home/dl/VSST/dm/novelty-detection-master/models/base.py", line 33, in call
return super(BaseModule, self).call(*args, **kwargs)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/dl/VSST/dm/novelty-detection-master/models/blocks_3d.py", line 133, in forward
activation_fn=self._activation_fn
File "/home/dl/VSST/dm/novelty-detection-master/models/blocks_3d.py", line 33, in residual_op
ha = f1(ha)
File "/home/dl/VSST/dm/novelty-detection-master/models/base.py", line 33, in call
return super(BaseModule, self).call(*args, **kwargs)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/dl/VSST/dm/novelty-detection-master/models/layers/mconv3d.py", line 29, in forward
return super(MaskedConv3d, self).forward(x)
File "/home/dl/anaconda3/envs/pytorch0.4/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 421, in forward
self.padding, self.dilation, self.groups)
RuntimeError: Expected 5-dimensional input for 5-dimensional weight [8, 1, 3, 3, 3], but got input of size [105, 690, 1, 8, 32, 32] instead

[DavideA]

Hi,

I think you are not using the correct collate function in the DataLoader.
One quick way to patch this is to reshape the input tensor before feeding the network:

x_in = x_in.view(-1, *x_in.shape[2:])

Hope this helps,
D

[DengMiao55]

@DavideA thank you，it‘s useful。 but I can't use the batch_size mentioned in the paper 。Do I need to use multiple GPUs？

[DavideA]

I used a single 12GB GPU.
I'd say you could reduce the batch size and give it a try.

D

[DengMiao55]

@DavideA Hi，I try to recurring your result in cifar ，but I only get AUC 0.6. could you tell me more about the detail about training。 thanks

[liuem607]

Hi, I have a similar question regarding input size. If my dataloader works correctly, the input data size before patching as you mentioned is [1, 690, 1, 8, 32, 32]. If I understand correctly, this first 1 is batch_size, and 8 is 8 sequential frames, and 32*32 is resized frame shape. But what is 690?

Also, the original image size is 256384, isn't 3232 too small?

Thank you in advance!

[ColaWithIce]

@DavideA Hi，I try to recurring your result in cifar ，but I only get AUC 0.6. could you tell me more about the detail about training。 thanks

Hi, where did you get the training code?

[shen-yao]

@DavideA Hi，I try to recurring your result in cifar ，but I only get AUC 0.6. could you tell me more about the detail about training。 thanks

Hi, where did you get the training code?

Same question please

[BenedictGreen]

Thx for the great work! I got same question about the details about training. hoping for author uploading the training code.@DavideA
thanks again!