Smart Dev

Multi Class Training not working.

11

RuntimeError: Error(s) in loading state_dict for RetinaNet: size mismatch for head.classification_head.cls_logits.weight: copying a param with shape torch.Size([9, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([18, 256,...

Thanks for your good project. I saw your project, Very excellent project. But that was built with swift 4 version. How can I get swift 5 version code?