forget add map_location for torch.load

m-tassano / fastdvdnet

FastDVDnet: A Very Fast Deep Video Denoising algorithm

MIT License

586 stars 125 forks source link

forget add map_location for torch.load #23

Closed suitmyself closed 3 years ago

suitmyself commented 3 years ago

In test_fastdvdnet.py , please change state_temp_dict = torch.load(args['model_file']) to state_temp_dict = torch.load(args['model_file'], map_location=device)

m-tassano commented 3 years ago

done. thanks for the suggestion