mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
updated test docs
This commit is contained in:
@@ -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),
|
||||
|
||||
@@ -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__':
|
||||
|
||||
Reference in New Issue
Block a user