(data, label)
| 67 | |
| 68 | @trace |
| 69 | def train_fun(data, label): |
| 70 | with gm: |
| 71 | net.train() |
| 72 | pred = net(data) |
| 73 | loss = F.nn.cross_entropy(pred, label) |
| 74 | gm.backward(loss) |
| 75 | return pred, loss |
| 76 | |
| 77 | @trace |
| 78 | def val_fun(data, label): |
no test coverage detected