search

mzbac / sketch2code

a simple model that implemented sketch to code
116 stars 40 forks source link

where to load pretrained weights ? #2

Open Nikkeii opened 5 years ago

Nikkeii commented 5 years ago

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 commented 4 years ago

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 commented 3 years ago

error how to solve these error of image