diff --git a/cutie/model/trainer.py b/cutie/model/trainer.py index ffbfc0b..ddce5db 100644 --- a/cutie/model/trainer.py +++ b/cutie/model/trainer.py @@ -41,7 +41,7 @@ def __init__(self, cfg: DictConfig, stage_cfg: DictConfig, log: TensorboardLogge self.log.log_string('model_size', str(sum([param.nelement() for param in self.cutie.parameters()]))) self.log.log_string( - 'number_of_parameters_that_requires_gradient', + 'number_of_parameters_that_require_gradient', str( sum([ param.nelement()