facebookresearch / facebookresearch/SlowFast

Model Loading on Pytorch

Open
#513 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.4k
Forks
1.3k
PR merge metrics
No merged PRs in 30d

Description

### Backstory
I have to load and run the models using pytorch. I dont have to run the direct model on the kinetics dataset, instead i have to tweak some layes, and then rerun it for some other task.

So to load the model in pytorch. Could you help with the same.
I tried loading using torch.load but that fails i guess, because the models have originally been saved through the caffe2 framework

I read all the other issues on this, so all the solutions include just changing the checkpointtype to caffe2 in the terminal command statement.
But the run_net.py is basically to ((frame model, load checkpoint, and run on data))
whereas the usecase i want is just to load the model into my python code and use it just another pytorch model.
I just want to load the caffe2 model, load its checkpoints into it, and just save it as a .pt model file, that i can use later.

### The main question
Can anyone help me with steps, to do the same.
I just need to load the pretrained models, in my pytorch python code and then run is accordingly later on.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.