from local
This commit is contained in:
parent
fef51784e4
commit
b0f43db3b9
Binary file not shown.
@ -1057,6 +1057,7 @@ def train_one_epoch(
|
||||
)
|
||||
|
||||
print(loss_info)
|
||||
exit()
|
||||
try: loss_info.reduce(loss.device)
|
||||
except: pass
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user