loss = criterion(outputs.float(), targets.long())