From e7b349a4cf93951a042777fa9eabb24f0b073d62 Mon Sep 17 00:00:00 2001 From: Varuna Jayasiri Date: Mon, 19 Oct 2020 20:36:44 +0530 Subject: [PATCH] tracker loss --- labml_nn/sketch_rnn/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/labml_nn/sketch_rnn/__init__.py b/labml_nn/sketch_rnn/__init__.py index e6cf9bfe..74330370 100644 --- a/labml_nn/sketch_rnn/__init__.py +++ b/labml_nn/sketch_rnn/__init__.py @@ -357,7 +357,7 @@ class StrokesBatchStep(BatchStepProtocol): hook_model_outputs(self.encoder, 'encoder') hook_model_outputs(self.decoder, 'decoder') - tracker.set_scalar("loss.*", True) + tracker.set_scalar("loss.total.*", True) def prepare_for_iteration(self): if MODE_STATE.is_train: