Kodar11 comments

Repositories
Issues
Comments

Results 1 comments of


                                            Kodar11

How to load pretrain model manually in correct way?

These error popping "'collections.OrderedDict' object has no attribute 'to' " when I am passing loaded train model for predications ''' loaded_model_1 = torch.load(r'/content/drive/MyDrive/models/Model_0') pred_and_plot_image(model=loaded_model_1, image_path=r'/content/drive/MyDrive/2 B/Training/destroyedbuilding/8.jpeg', class_names=class_names, # transform=weights.transforms(), #...