updated test docs

This commit is contained in:
William Falcon
2019-07-24 10:30:41 -04:00
parent 490da9f7d3
commit 57a99e2aa5
2 changed files with 4 additions and 0 deletions
+2
View File
@@ -254,6 +254,8 @@ class Trainer(TrainerIO):
@property
def __tng_tqdm_dic(self):
import pdb
pdb.set_trace()
tqdm_dic = {
'tng_loss': '{0:.3f}'.format(self.avg_loss),
'v_nb': '{}'.format(self.experiment.version),
+2
View File
@@ -60,6 +60,8 @@ def main():
# correct result and ok accuracy
assert result == 1, 'amp + ddp model failed to complete'
trainer.tng_tqdm_dic
clear_tt_dir()
if __name__ == '__main__':