parent
d3dceddf31
commit
77e88844cd
|
@ -10,7 +10,7 @@ model = dict(
|
||||||
loss=dict(type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0)))
|
loss=dict(type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0)))
|
||||||
|
|
||||||
# load model pretrained on imagenet
|
# 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
|
||||||
optimizer = dict(
|
optimizer = dict(
|
Loading…
Reference in New Issue