Paul-Edouard Sarlin commited on
Commit
09f8fa3
·
unverified ·
1 Parent(s): 0358e39

Add missing cast to Path (#31)

Browse files
Files changed (1) hide show
  1. 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)