Hello,
I am new to PyTorch and I am trying to load the MoCo model in order to use it.
In the following site: https://github.com/facebookresearch/moco I have found the code and also, I downloaded the pre-trained model (‘moco_v2_800ep_pretrain.pth.tar’) which is a state_dict with the model’s wheights.
I know that in order to use a model I need to create a model instance first and then load the state_dict.
My problem is that I can not create an instance model of MoCo. I saw the code on GiHub and I end up with the following code:
moco796×118 5.2 KB
I do not know what to put in the base_encoder parameter or tell me if there is another way to load the model.
Can anyone help me please?