learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-11-01 20:28:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

♻️ experiment.configs

Browse Source
This commit is contained in:
Varuna Jayasiri
2020-11-14 10:09:27 +05:30
parent ea976ab790
commit a331c99bdc
5 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
labml_nn/transformers/knn/build_index.py
View File

@ -33,7 +33,7 @@ def load_experiment(run_uuid: str, checkpoint: Optional[int] = None):
# This experiment is just an evaluation; i.e. nothing is tracked or saved
experiment.evaluate()
# Initialize configurations
experiment.configs(conf, conf_dict, 'run')
experiment.configs(conf, conf_dict)
# Set models for saving/loading
experiment.add_pytorch_models(get_modules(conf))
# Specify the experiment to load from

5
labml_nn/transformers/knn/train_model.py
View File

@ -331,10 +331,7 @@ def main():
'transformer.d_model': 256,
'transformer.d_ff': 1024,
'transformer.n_heads': 8,
'transformer.n_layers': 6},
# We need to load the function `TrainValidConfigs.run` and
# everything that it's dependent on
'run')
'transformer.n_layers': 6})
# Set models for saving and loading
experiment.add_pytorch_models(get_modules(conf))