I am trying to fine tune a YOLO network by following the SSD fine tuning example on the Gluon tutorial, by simply changing all the SSD functions to the equivalent Yolo functions, however I get the following error:
RuntimeError: Parameter ‘yolov319_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
This error arises from the line: _, _, anchors = net(mx.nd.zeros((1, 3, height, width))) in the get_dataloader function.
What else do I need to change to make the SSD script work for YOLO?