github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Updated some explanations

pull/825/head
gongjy 2024-09-20 17:05:50 +08:00
parent beca801dc2
commit aabe6566bd
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
2-eval.py
View File

@ -55,7 +55,6 @@ def setup_seed(seed):
torch.backends.cudnn.deterministic = True # 确保每次返回的卷积算法是确定的 torch.backends.cudnn.deterministic = True # 确保每次返回的卷积算法是确定的
torch.backends.cudnn.benchmark = False # 关闭 cuDNN 的自动调优，避免不确定性 torch.backends.cudnn.benchmark = False # 关闭 cuDNN 的自动调优，避免不确定性
if __name__ == "__main__": if __name__ == "__main__":
# ----------------------------------------------------------------------------- # -----------------------------------------------------------------------------
out_dir = 'out' out_dir = 'out'