mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 10:44:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix diagnostics position

Browse Source
This commit is contained in:
yaozengwei 2023-08-14 14:50:31 +08:00
parent d9bff38eea
commit 7291021a57
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/imagenet/CLS/swin_transformer/train.py
View File

@ -874,6 +874,10 @@ def run(rank, world_size, args):
rank=rank,
)
if params.print_diagnostics:
diagnostic.print_diagnostics()
break
validate(
params=params,
model=model,
@ -882,10 +886,6 @@ def run(rank, world_size, args):
tb_writer=tb_writer,
)
if params.print_diagnostics:
diagnostic.print_diagnostics()
break
save_checkpoint(
params=params,
model=model,