From 03ccce3f9cf0176a2a2b4ff3eeb11cbf22ef8f4b Mon Sep 17 00:00:00 2001
From: Jirka <jirka.borovec@seznam.cz>
Date: Sat, 17 Apr 2021 20:37:08 +0200
Subject: [PATCH] v0.2.3

---
 CHANGELOG.md             | 11 +++++++++++
 flash/__init__.py        |  2 +-
 tests/core/test_model.py |  2 +-
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 07d0d61e0d..ebcd5ac589 100644
--- 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
index e46a441eea..5aa9abc6de 100644
--- 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__ = "name@pytorchlightning.ai"
 __license__ = 'Apache-2.0'
diff --git a/tests/core/test_model.py b/tests/core/test_model.py
index 823243a132..9918dbb865 100644
--- 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)