a791472 0ba62b0 a791472
1
2
3
4
5
6
7
8
9
10
11
<pre> for batch in dataloader: optimizer.zero_grad() inputs, targets = batch inputs = inputs.to(device) targets = targets.to(device) outputs = model(inputs) loss = loss_function(outputs, targets) loss.backward() optimizer.step() scheduler.step()</pre>
