fix: correct 'complied' typo and replace print with logging
Apply changes from PR #396 by @shahrukhx01: - Fix typo 'complied' -> 'compiled' in ForecastConfig docstrings - Replace bare print() with logging.info() in load_checkpoint()
This commit is contained in:
@@ -23,11 +23,11 @@ class ForecastConfig:
|
|||||||
"""Options for forecasting.
|
"""Options for forecasting.
|
||||||
|
|
||||||
Attributes:
|
Attributes:
|
||||||
max_context: The maximum context length. This is used by the complied decode
|
max_context: The maximum context length. This is used by the compiled decode
|
||||||
function at inference time during batched inference. Any input time series
|
function at inference time during batched inference. Any input time series
|
||||||
with length less than max_context will be padded with zeros, and with
|
with length less than max_context will be padded with zeros, and with
|
||||||
length greater than max_context will be truncated.
|
length greater than max_context will be truncated.
|
||||||
max_horizon: The maximum horizon length. This is used by the complied decode
|
max_horizon: The maximum horizon length. This is used by the compiled decode
|
||||||
function at inference time during batched inference. The compiled cached
|
function at inference time during batched inference. The compiled cached
|
||||||
decoding function will by default forecast till max_horizon.
|
decoding function will by default forecast till max_horizon.
|
||||||
normalize_inputs: Whether to normalize the inputs. This is useful when the
|
normalize_inputs: Whether to normalize the inputs. This is useful when the
|
||||||
|
|||||||
@@ -85,7 +85,7 @@ class TimesFM_2p5_200M_torch_module(nn.Module):
|
|||||||
if "torch_compile" in kwargs:
|
if "torch_compile" in kwargs:
|
||||||
torch_compile = kwargs["torch_compile"]
|
torch_compile = kwargs["torch_compile"]
|
||||||
if torch_compile:
|
if torch_compile:
|
||||||
print("Compiling model...")
|
logging.info("Compiling model...")
|
||||||
self = torch.compile(self)
|
self = torch.compile(self)
|
||||||
|
|
||||||
self.eval()
|
self.eval()
|
||||||
|
|||||||
Reference in New Issue
Block a user