Hi, I’m trying to fine tune YOLO3 using the SSD example given in https://gluon-cv.mxnet.io/build/examples_detection/finetune_detection.html#sphx-glr-download-build-examples-detection-finetune-detection-py
I only make minor changes to that code (SSD to YOLO3). So I use:
net = gcv.model_zoo.get_model(‘yolo3_darknet53_voc’, pretrained=True)
net.reset_class(classes)
and in the following function I change SSDDefaultTrainTransform to YOLO3DefaultTrainTransform:
def get_dataloader(net, train_dataset, data_shape, batch_size, num_workers):
from gluoncv.data.transforms.presets.yolo import YOLO3DefaultTrainTransform
with autograd.train_mode():
…
_, _, anchors = net(mx.nd.zeros((1, 3, height, width)))
…
I get the following error:
RuntimeError: Parameter ‘yolov35_yolooutputv30_conv0_weight’ has not been initialized. Note that you should initialize parameters and create Trainer with Block.collect_params() instead of Block.params because the later does not include Parameters of nested child Blocks
Any suggestions on how I can fix this error?
Thank you.