parent
d3dceddf31
commit
77e88844cd
|
@ -10,7 +10,7 @@ model = dict(
|
|||
loss=dict(type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0)))
|
||||
|
||||
# load model pretrained on imagenet
|
||||
load_from = 'https://download.openmmlab.com/mmclassification/v0/vgg/vgg16_imagenet-91b6d117.pth' # noqa
|
||||
load_from = 'https://download.openmmlab.com/mmclassification/v0/vgg/vgg16_batch256_imagenet_20210208-db26f1a5.pth' # noqa
|
||||
|
||||
# optimizer
|
||||
optimizer = dict(
|
Loading…
Reference in New Issue