2019.4.24 An function has been writed to load the pretrained model
which was trained on imagenet-1k.The project of training the backbone
can be Downloaded from here -https://github.com/huaifeng1993/ILSVRC2012.
Limited to my computing resources(only have one RTX2080),I trained the
backbone on ILSVRC2012 with only 22 epochs.But it have a great impact on
the results.
由于我们没有ILSVRC2012的pretrained model,所以需要关掉标志位:
1
net = dfanet(pretrained=False, num_classes=20)
ERROR:
TypeError: init() got an unexpected keyword argument
'log_dir'
打开train.py,修改为:
1
writer = SummaryWriter(logdir=self.log_dir)
ERROR:
Unexpected bus error encountered in worker. This might be caused by
insufficient shared memory (shm)