Skip to content
This repository has been archived by the owner on Apr 3, 2023. It is now read-only.

where to load pretrained weights ? #2

Open
Nikkeii opened this issue Nov 1, 2019 · 2 comments
Open

where to load pretrained weights ? #2

Nikkeii opened this issue Nov 1, 2019 · 2 comments

Comments

@Nikkeii
Copy link

Nikkeii commented Nov 1, 2019

We have successfully run the docker containers but we don't know how to load pre-trained weights
encoder = torch.load('model_weights/encoder_resnet34_0.061650436371564865.pt')
decoder = torch.load('model_weights/decoder_resnet34_0.061650436371564865.pt')
what is use of this and where to insert it??

@rainmaker29
Copy link

We have successfully run the docker containers but we don't know how to load pre-trained weights
encoder = torch.load('model_weights/encoder_resnet34_0.061650436371564865.pt')
decoder = torch.load('model_weights/decoder_resnet34_0.061650436371564865.pt')
what is use of this and where to insert it??

They're used if u cannot train the model on your own. Insert them in your code after builiding the model and just before starting the training phase.

@vishalghag
Copy link

error
how to solve these error of
image

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

No branches or pull requests

3 participants