Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Share ckpt.bin #10

Open
kyelok opened this issue Dec 5, 2019 · 4 comments
Open

Share ckpt.bin #10

kyelok opened this issue Dec 5, 2019 · 4 comments

Comments

@kyelok
Copy link

kyelok commented Dec 5, 2019

Would you be able to share your ckpt.bin pre-trained weights?
ckpt.t7 from https://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6 doesn't seem to be the same format as yours.

@shodow
Copy link

shodow commented Dec 18, 2019

+1

1 similar comment
@qzp-user
Copy link

+1

@shodow
Copy link

shodow commented Dec 25, 2019

according to #6 and #5 , i use this code to convert ckpt.t7 to ckpt.bin

import os
import torch
import numpy as np

model_path="./ckpt.t7"
state_dict = torch.load(model_path)['net_dict']
with open("ckpt.bin", "wb") as f:
    for key in state_dict.keys():
        f.write(state_dict[key].cpu().numpy())

@kumarvis
Copy link

Is it possible to get your feature extractor network training code in case we want to use custome object.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants