forget add map_location for torch.load

[suitmyself]

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]

done.
thanks for the suggestion