learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-29 09:38:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

tracker loss

Browse Source
This commit is contained in:
Varuna Jayasiri
2020-10-19 20:36:44 +05:30
parent 28c800c613
commit e7b349a4cf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
labml_nn/sketch_rnn/__init__.py
View File

@ -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: