We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 832852a commit e4369ebCopy full SHA for e4369eb
1 file changed
internnav/trainer/navdp_trainer.py
@@ -200,6 +200,6 @@ def save_model(self, output_dir, state_dict=None, **kwargs):
200
201
# ensure the output directory exists
202
os.makedirs(output_dir, exist_ok=True)
203
- torch.save(model_to_save.state_dict(), output_dir + "navdp.ckpt")
+ torch.save(model_to_save.state_dict(), os.path.join(output_dir + "navdp.ckpt"))
204
205
print(f"Saving model to {output_dir} (is DDP: {hasattr(self.model, 'module')})")
0 commit comments