From c0fdfabaf32a3a5c7bda1d04a7263b7aa7a268dd Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 19 May 2022 11:30:56 +0800 Subject: [PATCH] Remove memory-limit options arg --- icefall/diagnostics.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/icefall/diagnostics.py b/icefall/diagnostics.py index d0a0461d0..1cd685d37 100644 --- a/icefall/diagnostics.py +++ b/icefall/diagnostics.py @@ -28,16 +28,12 @@ class TensorDiagnosticOptions(object): """Options object for tensor diagnostics: Args: - memory_limit: - The maximum number of bytes per tensor - (limits how many copies of the tensor we cache). max_eig_dim: The maximum dimension for which we print out eigenvalues (limited for speed reasons). """ - def __init__(self, memory_limit: int = (2 ** 20), max_eig_dim: int = 512): - self.memory_limit = memory_limit + def __init__(self, max_eig_dim: int = 512): self.max_eig_dim = max_eig_dim def dim_is_summarized(self, size: int):