Closed XWalways closed 5 years ago
train on DOTA num_class=15，there is an error InvalidArgumentError:Assign requires shapes of both tensors to match. lhs shape= [16] rhs shape= [2] Maybe the pretrained model was trained on your own dataset instead of ImageNet?
train on DOTA num_class=15，there is an error InvalidArgumentError:Assign requires shapes of both tensors to match. lhs shape= [16] rhs shape= [2] Maybe the pretrained model was trained on your own dataset instead of ImageNet?