Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
hkchengrex authored Jun 24, 2024
1 parent e869fab commit a9cad9a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cutie/model/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit a9cad9a

Please sign in to comment.