From fd4daa0494a2e76c52ca76311dcfe30f140bd841 Mon Sep 17 00:00:00 2001
From: criselsuarez <20343857+criselsuarez@users.noreply.github.com>
Date: Wed, 19 Oct 2022 03:28:38 -1000
Subject: [PATCH] changing the num_heads from None to 0

---
 zoobot/tensorflow/estimators/maxvit_zoobot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/zoobot/tensorflow/estimators/maxvit_zoobot.py b/zoobot/tensorflow/estimators/maxvit_zoobot.py
index 3f56d036..d2e4b17a 100644
--- a/zoobot/tensorflow/estimators/maxvit_zoobot.py
+++ b/zoobot/tensorflow/estimators/maxvit_zoobot.py
@@ -123,7 +123,7 @@ class Attention(tf.keras.layers.Layer):
   def __init__(self,
                hidden_size,
                head_size,
-               num_heads=None,
+               num_heads=0,
                dropatt=0.0,
                attn_axis=0,
                rel_attn_type=None,