OMEGA FIX
This commit is contained in:
parent
6c5991d5bb
commit
7a1444e533
@ -24,7 +24,7 @@
|
||||
]
|
||||
]
|
||||
},
|
||||
"heartbeat": "2023-05-11T19:49:33.658443",
|
||||
"heartbeat": "2023-05-11T19:53:53.820775",
|
||||
"host": {
|
||||
"ENV": {},
|
||||
"cpu": "Intel(R) Core(TM) i7-6700K CPU @ 4.00GHz",
|
||||
|
@ -47,9 +47,9 @@ def trainNet(_run, trainloader, criterion, optimizer, net, epochs=20):
|
||||
outputs = net(inputs)
|
||||
|
||||
loss = criterion(outputs, labels)
|
||||
print(loss.item())
|
||||
|
||||
_run.log_scalar("training.loss", loss)
|
||||
_run.log_scalar("training.loss", loss.item())
|
||||
|
||||
loss.backward()
|
||||
optimizer.step()
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user