RuntimeError while training

[VyBui]

Hello @PeikeLi,

I cloned your train code and config to my own dataset like LIP:

However, when I run python train.py this is what I have got:

I wanted to ask if this is my mistake?

[GoGoDuck912]

I suppose that you are training with your own dataset. By default, pytorch takes the input dimensions as NCHW. Check it by yourself.

[VyBui]

@PeikeLi thank you for you reply,

Yes, I am training with my own dataset, the error was the order of NCHW. I got the training running will see if it works.