learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-27 20:24:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix checkpoint path

Browse Source
This commit is contained in:
Varuna Jayasiri
2022-09-13 17:07:34 +05:30
committed by GitHub
parent a5686f4709
commit ee97a5ff48
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
labml_nn/neox/checkpoint.py
View File

@ -36,6 +36,8 @@ def get_checkpoints_download_path():
_CHECKPOINTS_DOWNLOAD_PATH = lab.get_data_path() / 'neox' / 'slim_weights'
inspect(neox_checkpoint_path=_CHECKPOINTS_DOWNLOAD_PATH)
return _CHECKPOINTS_DOWNLOAD_PATH
def get_files_to_download(n_layers: int = 44):
"""