Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add pytorch version reporting

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
map2map/test.py
View File

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

1
map2map/train.py
View File

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