"src/array/cuda/vscode:/vscode.git/clone" did not exist on "9a00cf194fcf994b2527cd927d691144f5e9c47b"
Commit 3f0a9b8d authored by wanglch's avatar wanglch
Browse files

Update single_dcu_train.py

parent dcb1010d
...@@ -202,7 +202,7 @@ if __name__=='__main__': ...@@ -202,7 +202,7 @@ if __name__=='__main__':
# 获取当前的日期和时间 # 获取当前的日期和时间
now = datetime.now() now = datetime.now()
timestamp = now.strftime("%Y%m%d_%H%M%S") timestamp = now.strftime("%Y%m%d_%H%M%S")
new_model_path = f'saves/umt5_{timestamp}' new_model_path = f'./saves/umt5_{timestamp}'
model.module.save_pretrained(new_model_path) model.module.save_pretrained(new_model_path)
tokenizer.save_pretrained(new_model_path) tokenizer.save_pretrained(new_model_path)
print("Done!") print("Done!")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment