RoCL
RoCL copied to clipboard
pretrained model using
how can i use a pretrained model at first? your model is different from normal resnet networks, do you train any model as pretrained model and how can I git it
I think your error is related with this line.
model.load_state_dict(new_state_dict)
I changed the above line to below.
model.load_state_dict(new_state_dict, strict=False)