Add pytorch version reporting

This commit is contained in:
Yin Li 2020-09-13 16:14:48 -04:00
parent a579c9655b
commit b7a9adf0c7
2 changed files with 2 additions and 0 deletions

View File

@ -11,6 +11,7 @@ from .utils import import_attr, load_model_state_dict
def test(args):
print('pytorch {}'.format(torch.__version__))
pprint(vars(args))
sys.stdout.flush()

View File

@ -184,6 +184,7 @@ def gpu_worker(local_rank, node, args):
logger = SummaryWriter()
if rank == 0:
print('pytorch {}'.format(torch.__version__))
pprint(vars(args))
sys.stdout.flush()