search

princeton-vl / RAFT

BSD 3-Clause "New" or "Revised" License
3.12k stars 621 forks source link

Error while training with flying things 3d #139

Open abhijit0 opened 2 years ago

abhijit0 commented 2 years ago

I am getting the following error for the line 172, flow_predictions = model(image1, image2, iters=args.iters)

The error is : TypeError: forward() missing 2 required positional arguments: 'image1' and 'image2' Although the images are loaded correctly, I am getting this error.

Geniussh commented 2 years ago

Are you using multi gpus? If so, I met this before because I accidentally forgot to increase the batch size proportionally when I increased one extra gpu so that it became indivisible. Make sure you batch size is divisible by the number of gpus.