Error during training - Githubissues

fanq15 / FSOD-code

MIT License

244 stars 61 forks source link

Error during training #40

Open YRDTSS opened 4 years ago

YRDTSS commented 4 years ago

Error display during training: "Expected 4-dimensional input for 4-dimensional weight [64, 3, 7, 7], but got input of size [4, 10, 3, 320, 320] instead". I don't know why this error was reported.

fanq15 commented 4 years ago

You should set 1 image in 1 batch. If you only have 1 GPU, please set bs=1.

YRDTSS commented 4 years ago

Thank you for your reply. It is very helpful.