From 69ce218f14919da2d4f4611d2af2bb2375dbca7b Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Mon, 24 Jul 2023 16:29:04 +0200
Subject: [PATCH] Fix issue on var name
---
osrt/layers.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/osrt/layers.py b/osrt/layers.py
index 19f9395..fbec560 100644
--- a/osrt/layers.py
+++ b/osrt/layers.py
@@ -244,7 +244,7 @@ class SlotAttention(nn.Module):
k, v = self.to_k(inputs), self.to_v(inputs)
- slots = self.slots_mu + torch.exp(self.slots_log_sigma) * torch.randn(len(inputs), self.num_slots, self.slot_size, device = self.slots_mu.device)
+ slots = self.slots_mu + torch.exp(self.slots_log_sigma) * torch.randn(len(inputs), self.num_slots, self.slot_dim, device = self.slots_mu.device)
# Multiple rounds of attention.
for _ in range(self.iters):
--
GitLab