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-29 17:57:14 +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

3
labml_nn/capsule_networks/mnist.py
View File

@ -153,8 +153,7 @@ def main():
conf = Configs() conf = Configs()
experiment.create(name='mnist_latest') experiment.create(name='mnist_latest')
experiment.configs(conf, {'optimizer.optimizer': 'Adam', experiment.configs(conf, {'optimizer.optimizer': 'Adam',
'device.cuda_device': 1}, 'device.cuda_device': 1})
'run')
with experiment.start(): with experiment.start():
conf.run() conf.run()

3
labml_nn/gan/dcgan.py
View File

@ -106,8 +106,7 @@ def main():
experiment.configs(conf, experiment.configs(conf,
{'discriminator': 'cnn', {'discriminator': 'cnn',
'generator': 'cnn', 'generator': 'cnn',
'label_smoothing': 0.01}, 'label_smoothing': 0.01})
'run')
with experiment.start(): with experiment.start():
conf.run() conf.run()

3
labml_nn/gan/simple_mnist_experiment.py
View File

@ -233,8 +233,7 @@ def main():
conf = Configs() conf = Configs()
experiment.create(name='mnist_gan', comment='test') experiment.create(name='mnist_gan', comment='test')
experiment.configs(conf, experiment.configs(conf,
{'label_smoothing': 0.01}, {'label_smoothing': 0.01})
'run')
with experiment.start(): with experiment.start():
conf.run() conf.run()

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 # This experiment is just an evaluation; i.e. nothing is tracked or saved
experiment.evaluate() experiment.evaluate()
# Initialize configurations # Initialize configurations
experiment.configs(conf, conf_dict, 'run') experiment.configs(conf, conf_dict)
# Set models for saving/loading # Set models for saving/loading
experiment.add_pytorch_models(get_modules(conf)) experiment.add_pytorch_models(get_modules(conf))
# Specify the experiment to load from # 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_model': 256,
'transformer.d_ff': 1024, 'transformer.d_ff': 1024,
'transformer.n_heads': 8, 'transformer.n_heads': 8,
'transformer.n_layers': 6}, 'transformer.n_layers': 6})
# We need to load the function `TrainValidConfigs.run` and
# everything that it's dependent on
'run')
# Set models for saving and loading # Set models for saving and loading
experiment.add_pytorch_models(get_modules(conf)) experiment.add_pytorch_models(get_modules(conf))