Closed danielcrane closed 3 years ago
Does anyone know if there's any easy way to resume training from for example weights/SOLOv2_R50_3x.pth , however with a different number of classes?
weights/SOLOv2_R50_3x.pth
In my case I only have one output class, however of course if I simply set num_classes=2 I get a tensor size mismatch error.
num_classes=2
Setting load_from= in the config file resolved this.
load_from=
Does anyone know if there's any easy way to resume training from for example
weights/SOLOv2_R50_3x.pth
, however with a different number of classes?In my case I only have one output class, however of course if I simply set
num_classes=2
I get a tensor size mismatch error.