jfrancis99

Multiple issues to get running:

3

1) function run_batch in train.py ``` def run_batch(model, args, device, compute_loss_fct): for arg in args: if arg is not None: arg = arg.to(device) output = model(*args, device=device) allloss = compute_loss_fct(output,...