Skip to content

Commit

Permalink
change ckpt_prefix to relative path
Browse files Browse the repository at this point in the history
Before, `ckpt_prefix` is an absolute path and the generated symilnks link to an abosulote path. If users wants to move the whole directory, the relative path should be more user-friendly.

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
  • Loading branch information
njzjz committed Aug 4, 2023
1 parent 4fa54ec commit 611770b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmd/train/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -811,7 +811,7 @@ def save_checkpoint(self, cur_batch: int):
try:
ckpt_prefix = self.saver.save(
self.sess,
os.path.join(os.getcwd(), self.save_ckpt),
self.save_ckpt,
global_step=cur_batch,
)
except google.protobuf.message.DecodeError as e:
Expand Down

0 comments on commit 611770b

Please sign in to comment.