OMEGA FIX

This commit is contained in:
Jakub Henyk 2023-05-11 21:53:55 +02:00
parent 6c5991d5bb
commit 7a1444e533
2 changed files with 3 additions and 3 deletions

View File

@ -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",

View File

@ -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()