refinedet.pytorch icon indicating copy to clipboard operation
refinedet.pytorch copied to clipboard

About training

Open zeroorhero opened this issue 4 years ago • 2 comments

Hello,thank you for your code! when i am training ,there is no error .but can not train.

CUDA devices: <class 'torch.cuda.device'> GPU numbers: 1 Create or load an imdb. Loaded dataset voc_2007_trainval for training Preparing training data... voc_2007_trainval gt roidb loaded from /home/lcq/下载/residual-refinedet.pytorch-master/~/data/cache/voc_2007_trainval_gt_roidb.pkl done before filtering, there are 146 images... after filtering, there are 146 images... Construct vgg16_refinedet network. /home/lcq/下载/residual-refinedet.pytorch-master/libs/networks/refinedet.py:34: UserWarning: volatile was removed and now has no effect. Use with torch.no_grad(): instead. self.anchors = Variable(self.anchor_layer.forward(), volatile=True) Loading pretrained weights from weights/vgg16/vgg16_reducedfc.pth

/home/lcq/下载/residual-refinedet.pytorch-master/libs/networks/vgg_refinedet.py(56)_init_modules() -> assert key_layer_ids[2] == -1 and key_layer_ids[3] == -1,
(Pdb)

do you know why? thank you !

zeroorhero avatar Apr 05 '20 13:04 zeroorhero