pytorch-ssd icon indicating copy to clipboard operation
pytorch-ssd copied to clipboard

init_from_pretrained_ssd error

Open HongChow opened this issue 5 years ago • 0 comments

RuntimeError: Error(s) in loading state_dict for SSD: my command(using in spyder terminal): %run train_ssd.py --dataset_type voc --datasets ./data --net vgg16-ssd --pretrained_ssd models/vgg16_reducedfc.pth --batch_size 24 --freeze_net --num_epochs 200 --lr 0.001

it encounter an error on the line: self.load_state_dict(model_dict) the 'vgg16_reducedfc.pth ' was downloaded from the website provided by the official's (qfgaohao)

HongChow avatar Sep 05 '19 03:09 HongChow