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)
|