Spaces:

ashu316
/

tgn-playground

Build error

ashu316 commited on Apr 14

Commit

4e3dad4

verified ·

1 Parent(s): d1cb418

Update modules/memory_updater.py

Files changed (1) hide show

modules/memory_updater.py CHANGED Viewed

@@ -23,7 +23,8 @@ class SequenceMemoryUpdater(MemoryUpdater):
                                                                                      "update memory to time in the past"
     memory = self.memory.get_memory(unique_node_ids)
-    self.memory.last_update[unique_node_ids] = timestamps
     updated_memory = self.memory_updater(unique_messages, memory)

                                                                                      "update memory to time in the past"
     memory = self.memory.get_memory(unique_node_ids)
+    with torch.no_grad():
+        self.memory.last_update[unique_node_ids] = timestamps
     updated_memory = self.memory_updater(unique_messages, memory)