erorr about load ckpt
Opened this issue · 0 comments
Deleted user commented
resnet = models.resnet50(pretrained=False)
cfg = dict(
net=resnet,
image_size=224,
hidden_layer='avgpool',
projeion_size=256,
projection_hidden_size=4096,
moving_average_decay=0.99
)
model_C = SelfSupervisedLearner.load_from_checkpoint('./epoch=4-step=269.ckpt', **cfg)
👆this is my code. But I got this issue:
TypeError: init() got an unexpected keyword argument 'projeion_size'.
Do you know what happen?