Nkulkarni
Nkulkarni
@aprams Yes I did, had to do multiple changes to make that work. I basically flattened the mobilenetv2 model into a flat model before feeding into this code (dealing with...
I got the same error with Pytorch 0.4.1. > Training... Epoch-0-: 0%| | 0/391 [00:00
**Update:** worked for me when I used pytorch 0.3.0. The error no longer came up.
@funnyuday How did you finetune? I want to finetune the same model with my custom dataset as well, but I can't find out how to do that. Thanks in advance!