We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c6e7aa5 commit e1fc2f9Copy full SHA for e1fc2f9
cmonge/trainers/ot_trainer.py
@@ -199,7 +199,7 @@ def setup(
199
fitting_loss: Dict[str, Any],
200
regularizer: Dict[str, Any],
201
optim: Dict[str, Any],
202
- checkpointing_path: None # For compatability with base class
+ checkpointing_path: None, # For compatability with base class
203
) -> None:
204
"""Initializes models and optimizers."""
205
self.metrics["params"] = {
0 commit comments