lucidrains/byol-pytorch

erorr about load ckpt

Opened this issue · 0 comments

    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?