Updated readme and bumped version.

penguinmenac3 · Dec 20, 2021 · c6519b1 · c6519b1
1 parent 88192d0
commit c6519b1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -45,6 +45,7 @@ Here is the simplest mnist example, it is so short it can be part of the main re
 import torch
 from torch.optim import SGD
 
+from leanai.core.config import DictLike
 from leanai.core.experiment import Experiment, set_seeds
 from leanai.data.datasets import FashionMNISTDataset
 from leanai.training.losses import SparseCrossEntropyLossFromLogits
@@ -57,14 +58,14 @@ experiment = Experiment(
     output_path="outputs",
 )
 experiment.run_training(
-    load_dataset=dict(
+    load_dataset=DictLike(
         type=FashionMNISTDataset,
         data_path="outputs",
     ),
-    build_loss=dict(
+    build_loss=DictLike(
         type=SparseCrossEntropyLossFromLogits,
     ),
-    build_optimizer=dict(
+    build_optimizer=DictLike(
         type=SGD,
         lr=1e-3,
     ),

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 from codecs import open
 from os import path
 
-__version__ = '2.0.0'
+__version__ = '2.1.0'
 
 here = path.abspath(path.dirname(__file__))
 OFFICIAL_URL = "https://github.com/penguinmenac3/leanai"