fix hdfs path null

volcengine · Dec 18, 2024 · d1690d8 · d1690d8
1 parent 518f632
commit d1690d8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/verl/trainer/ppo/ray_trainer.py b/verl/trainer/ppo/ray_trainer.py
@@ -515,13 +515,15 @@ def fit(self):
                 if self.config.trainer.save_freq > 0 and (global_steps + 1) % self.config.trainer.save_freq == 0:
                     actor_local_path = os.path.join(self.config.trainer.default_local_dir, 'actor',
                                                     f'global_step_{global_steps}')
-                    actor_remote_path = os.path.join(self.config.trainer.default_hdfs_dir, 'actor')
+                    actor_remote_path = None if self.config.trainer.default_hdfs_dir is None else os.path.join(
+                        self.config.trainer.default_hdfs_dir, 'actor')
                     self.actor_rollout_wg.save_checkpoint(actor_local_path, actor_remote_path)
 
                     if self.use_critic:
                         critic_local_path = os.path.join(self.config.trainer.default_local_dir, 'critic',
                                                          f'global_step_{global_steps}')
-                        critic_remote_path = os.path.join(self.config.trainer.default_hdfs_dir, 'critic')
+                        critic_remote_path = None if self.config.trainer.default_hdfs_dir is None else os.path.join(
+                            self.config.trainer.default_hdfs_dir, 'critic')
                         self.critic_wg.save_checkpoint(critic_local_path, critic_remote_path)
 
                 global_steps += 1