From 51dd2adb31323c6499d00fc09d65d3272b9f8236 Mon Sep 17 00:00:00 2001
From: Abhirath Anand <74202102+theabhirath@users.noreply.github.com>
Date: Sat, 14 May 2022 19:40:57 +0530
Subject: [PATCH 1/2] Expose `nclasses` option for `DenseNet`

Previously, the `nclasses` option was not exposed for the `DenseNet` API with the `config` option.
---
 src/convnets/densenet.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/convnets/densenet.jl b/src/convnets/densenet.jl
index 53d96df09..4269b83d2 100644
--- a/src/convnets/densenet.jl
+++ b/src/convnets/densenet.jl
@@ -147,9 +147,9 @@ Set `pretrain = true` to load the model with pre-trained weights for ImageNet.
 
 See also [`Metalhead.densenet`](#).
 """
-function DenseNet(config::Int = 121; pretrain = false)
+function DenseNet(config::Int = 121; pretrain = false, nclasses = 1000)
   @assert config in keys(densenet_config) "`config` must be one out of $(sort(collect(keys(densenet_config))))."
-  model = DenseNet(densenet_config[config])
+  model = DenseNet(densenet_config[config]; nclasses = nclasses)
 
   pretrain && loadpretrain!(model, string("DenseNet", config))
   return model

From d8d0578c982bd61ac3f1207e85b33125ee321757 Mon Sep 17 00:00:00 2001
From: Abhirath Anand <74202102+theabhirath@users.noreply.github.com>
Date: Sun, 15 May 2022 00:51:11 +0530
Subject: [PATCH 2/2] Bump version

---
 Project.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Project.toml b/Project.toml
index 1d1eb4472..ac8973969 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "Metalhead"
 uuid = "dbeba491-748d-5e0e-a39e-b530a07fa0cc"
-version = "0.7.0"
+version = "0.7.1"
 
 [deps]
 Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"