Skip to content

Commit e4369eb

Browse files
committed
[fix] update the file path in saving navdp training ckpt
1 parent 832852a commit e4369eb

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

internnav/trainer/navdp_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,6 @@ def save_model(self, output_dir, state_dict=None, **kwargs):
200200

201201
# ensure the output directory exists
202202
os.makedirs(output_dir, exist_ok=True)
203-
torch.save(model_to_save.state_dict(), output_dir + "navdp.ckpt")
203+
torch.save(model_to_save.state_dict(), os.path.join(output_dir + "navdp.ckpt"))
204204

205205
print(f"Saving model to {output_dir} (is DDP: {hasattr(self.model, 'module')})")

0 commit comments

Comments
 (0)