v0.2.3

Lightning-Universe · Apr 17, 2021 · 03ccce3 · 03ccce3
1 parent c3f01d2
commit 03ccce3
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,17 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 
+## [0.2.3] - 2021-04-17
+
+### Added
+
+- Added TIMM integration as backbones ([#196](https://github.com/PyTorchLightning/lightning-flash/pull/196))
+
+### Fixed
+
+- Fixed `nltk.download` ([#210](https://github.com/PyTorchLightning/lightning-flash/pull/210))
+
+
 ## [0.2.2] - 2021-04-05
 
 ### Changed

diff --git a/flash/__init__.py b/flash/__init__.py
@@ -14,7 +14,7 @@
 """Root package info."""
 import os
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 __author__ = "PyTorchLightning et al."
 __author_email__ = "[email protected]"
 __license__ = 'Apache-2.0'

diff --git a/tests/core/test_model.py b/tests/core/test_model.py
@@ -44,7 +44,7 @@ def __len__(self) -> int:
 
 @pytest.mark.parametrize("metrics", [None, pl.metrics.Accuracy(), {"accuracy": pl.metrics.Accuracy()}])
 def test_classificationtask_train(tmpdir: str, metrics: Any):
-    model = nn.Sequential(nn.Flatten(), nn.Linear(28 * 28, 10), nn.LogSoftmax())
+    model = nn.Sequential(nn.Flatten(), nn.Linear(28 * 28, 10), nn.Softmax())
     train_dl = torch.utils.data.DataLoader(DummyDataset())
     val_dl = torch.utils.data.DataLoader(DummyDataset())
     task = ClassificationTask(model, F.nll_loss, metrics=metrics)