can we apply the transfer learning on this model ? if yes, how can we import the pre-trained model to modify the parameters .

[AmokraneIlhem]

[liuyuan-pal]

Hi, the model is stored as a standard pytorch ckpt file. You may refer to https://pytorch.org/tutorials/beginner/saving_loading_models.html for loading these parameters and pass them to the Optimizer for finetuning.

[AmokraneIlhem]

@liuyuan-pal
First of all, thank you for your answer. Where exactly is this file stored?

[liuyuan-pal]

These models are model_best.pth in the data folder.

[AmokraneIlhem]

@liuyuan-pal
so I have to refine each model (detector , viewpoint selector , refiner) ? I also have another question, please, how to freeze the layers of these models because I noticed that inside each model there are several methods. How do I just access the last layers? thank you in advance.