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

Remove memory-limit options arg

Browse Source
This commit is contained in:
Daniel Povey 2022-05-19 11:30:56 +08:00
parent c2c46ea023
commit c0fdfabaf3
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
icefall/diagnostics.py
View File

@ -28,16 +28,12 @@ class TensorDiagnosticOptions(object):
"""Options object for tensor diagnostics: """Options object for tensor diagnostics:
Args: Args:
memory_limit:
The maximum number of bytes per tensor
(limits how many copies of the tensor we cache).
max_eig_dim: max_eig_dim:
The maximum dimension for which we print out eigenvalues The maximum dimension for which we print out eigenvalues
(limited for speed reasons). (limited for speed reasons).
""" """
def __init__(self, memory_limit: int = (2 ** 20), max_eig_dim: int = 512): def __init__(self, max_eig_dim: int = 512):
self.memory_limit = memory_limit
self.max_eig_dim = max_eig_dim self.max_eig_dim = max_eig_dim
def dim_is_summarized(self, size: int): def dim_is_summarized(self, size: int):