notebook

Author: vpj

labml_nn/hypernetworks/experiment.py

@@ -92,7 +92,6 @@ def main():
     # Set models for saving and loading
     experiment.add_pytorch_models(get_modules(conf))
 
-    conf.init()
     # Start the experiment
     with experiment.start():
         # `TrainValidConfigs.run`

labml_nn/transformers/switch/__init__.py

@@ -32,8 +32,8 @@
 
 Here's [the training code](experiment.html) and a notebook for training a switch transformer on Tiny Shakespeare dataset.
 
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/lab-ml/nn/blob/master/labml_nn/transformers/feedback/experiment.ipynb)
-[![View Run](https://img.shields.io/badge/labml-experiment-brightgreen)](https://web.lab-ml.com/run?uuid=d8eb9416530a11eb8fb50242ac1c0002)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/lab-ml/nn/blob/master/labml_nn/transformers/switch/experiment.ipynb)
+[![View Run](https://img.shields.io/badge/labml-experiment-brightgreen)](https://web.lab-ml.com/run?uuid=c4656c605b9311eba13d0242ac1c0002)
 """
 
 import torch