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": {
|
"host": {
|
||||||
"ENV": {},
|
"ENV": {},
|
||||||
"cpu": "Intel(R) Core(TM) i7-6700K CPU @ 4.00GHz",
|
"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)
|
outputs = net(inputs)
|
||||||
|
|
||||||
loss = criterion(outputs, labels)
|
loss = criterion(outputs, labels)
|
||||||
print(loss.item())
|
|
||||||
|
|
||||||
_run.log_scalar("training.loss", loss)
|
_run.log_scalar("training.loss", loss.item())
|
||||||
|
|
||||||
loss.backward()
|
loss.backward()
|
||||||
optimizer.step()
|
optimizer.step()
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user