mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-04 06:34:20 +00:00
Incorporate PR https://github.com/k2-fsa/icefall/pull/1269
This commit is contained in:
parent
d5252a4157
commit
2d9d3d4995
@ -1180,7 +1180,7 @@ def run(rank, world_size, args):
|
||||
|
||||
if params.print_diagnostics:
|
||||
opts = diagnostics.TensorDiagnosticOptions(
|
||||
2**22
|
||||
512
|
||||
) # allow 4 megabytes per sub-module
|
||||
diagnostic = diagnostics.attach_diagnostics(model, opts)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user