diff --git a/my_runs/7/run.json b/my_runs/7/run.json index 472b4b0..da02255 100644 --- a/my_runs/7/run.json +++ b/my_runs/7/run.json @@ -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", diff --git a/sacred_train.py b/sacred_train.py index e0fd2e5..b0322b2 100644 --- a/sacred_train.py +++ b/sacred_train.py @@ -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()