From f92e98154a976c09b2da8731a4de3a9f8057cf9a Mon Sep 17 00:00:00 2001
From: Victor Joos <victor.joos@uclouvain.be>
Date: Tue, 28 Jun 2022 16:28:50 +0200
Subject: [PATCH] fix gernet

---
 hydra_flash/conf/model/gernet.yaml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hydra_flash/conf/model/gernet.yaml b/hydra_flash/conf/model/gernet.yaml
index 400426c..04e40fe 100644
--- a/hydra_flash/conf/model/gernet.yaml
+++ b/hydra_flash/conf/model/gernet.yaml
@@ -1,3 +1,6 @@
 _target_: flash.image.ImageClassifier
 backbone: gernet_s
 learning_rate: 1.e-3
+complete:
+  - labels
+  - multi_label
-- 
GitLab