Spaces:
Running
Running
Paul-Edouard Sarlin
commited on
Add missing cast to Path (#31)
Browse files- maploc/train.py +1 -1
maploc/train.py
CHANGED
|
@@ -120,7 +120,7 @@ def train(cfg: DictConfig, job_id: Optional[int] = None):
|
|
| 120 |
if init_checkpoint_path is not None:
|
| 121 |
logger.info("Initializing the model from checkpoint %s.", init_checkpoint_path)
|
| 122 |
model = GenericModule.load_from_checkpoint(
|
| 123 |
-
init_checkpoint_path, strict=True, find_best=False, cfg=cfg
|
| 124 |
)
|
| 125 |
else:
|
| 126 |
model = GenericModule(cfg)
|
|
|
|
| 120 |
if init_checkpoint_path is not None:
|
| 121 |
logger.info("Initializing the model from checkpoint %s.", init_checkpoint_path)
|
| 122 |
model = GenericModule.load_from_checkpoint(
|
| 123 |
+
Path(init_checkpoint_path), strict=True, find_best=False, cfg=cfg
|
| 124 |
)
|
| 125 |
else:
|
| 126 |
model = GenericModule(cfg)
|