Excuse me, error when running "python main.py --resume checkpoints/avobjects_loc_sep.pt --input_video demo.mp4 --output_dir demo_out"

[zhang123-sys]

python main.py --resume checkpoints/avobjects_loc_sep.pt --input_video demo.mp4 --output_dir demo_out Using device: cuda rm demo_out/* -rf Checkpoint checkpoints/avobjects_loc_sep.pt loaded! Resampling media/demo.mp4 to 25 fps /opt/anaconda3/envs/avobjects/lib/python3.6/site-packages/torchvision/transforms/functional.py:405: UserWarning: Argument interpolation should be of type InterpolationMode instead of int. Please, use InterpolationMode enum. "Argument interpolation should be of type InterpolationMode instead of int. " Traceback (most recent call last): File "main.py", line 44, in main() File "main.py", line 41, in main trainer.eval(dataloader) File "/data0/audio_project/github/avobjects/trainer.py", line 60, in eval mel, _, _, _ = wav2filterbanks(audio.to(self.device)) File "/data0/audio_project/github/avobjects/load_audio.py", line 42, in wav2filterbanks onesided=True) # b x F x T x 2 File "/opt/anaconda3/envs/avobjects/lib/python3.6/site-packages/torch/functional.py", line 573, in stft normalized, onesided, return_complex) RuntimeError: stft input and window must be on the same device but got self on cuda:0 and window on cpu

[pulkitmehtaworkmetacube]

I faced the same issue and i forced device to CPU in below code in trainer.py. class DemoEvalTrainer():

def __init__(self, model, opts):
    self.model = model
    self.opts = opts

    **self.device = torch.device('cpu')#torch.device('cuda:0')**