From e4369eb88fc955936ec51ecdd183dcd61ea12f89 Mon Sep 17 00:00:00 2001 From: wzcai99 Date: Fri, 6 Feb 2026 19:47:02 +0800 Subject: [PATCH] [fix] update the file path in saving navdp training ckpt --- internnav/trainer/navdp_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/internnav/trainer/navdp_trainer.py b/internnav/trainer/navdp_trainer.py index 5542562e..94d11095 100644 --- a/internnav/trainer/navdp_trainer.py +++ b/internnav/trainer/navdp_trainer.py @@ -200,6 +200,6 @@ def save_model(self, output_dir, state_dict=None, **kwargs): # ensure the output directory exists os.makedirs(output_dir, exist_ok=True) - 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")) print(f"Saving model to {output_dir} (is DDP: {hasattr(self.model, 'module')})")