missing param in train()--> turns out to be 'mode'

[toosi]

----- Training phase ----- Traceback (most recent call last): File "./src/benchmark.py", line 228, in main() File "./src/benchmark.py", line 219, in main args.niter, args.stepsize, losses, it=it) TypeError: train() missing 1 required positional argument: 'losses'

Fix:

while it < args.niter:
    it = train(
      device=device, loader=train_loader, model=net, mode=args.mode, optimizer=optimizer, 
      niter=args.niter, stepsize=args.stepsize, losses=losses, it=it)

Thank you for sharing your code.