Error while training with flying things 3d

[abhijit0]

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.

[SecureSheII]

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.