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-31 10:48:49 +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.encoder, 'encoder')
hook_model_outputs(self.decoder, 'decoder') hook_model_outputs(self.decoder, 'decoder')
tracker.set_scalar("loss.*", True) tracker.set_scalar("loss.total.*", True)
def prepare_for_iteration(self): def prepare_for_iteration(self):
if MODE_STATE.is_train: if MODE_STATE.is_train: