diff --git a/hydra_flash/conf/model/gernet.yaml b/hydra_flash/conf/model/gernet.yaml
index 400426c35f22ba6ebdbfc4995bf96b7e85e8d212..04e40fe1ebd1b0628d0692b9270c3cc6dcd91ac6 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