merged

atomicarchitects · Jul 27, 2024 · 60a91de · 60a91de
2 parents 5c01a0c + 7d5a152
commit 60a91de
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -0,0 +1,8 @@
+## TODO
+
+- [ ] Get `torch.export` pipeline working.
+- [ ] Train tetris
+- [ ] Port `allegro-jax`
+
+
+
diff --git a/model.py b/model.py
@@ -140,4 +140,4 @@ def forward(self,
                                     torch.zeros(num_nodes, dtype=torch.int64),
                                     dim=0,
                                     dim_size=1)
-        return self.readout_mlp(graph_globals)
+        return self.readout_mlp(graph_globals)