Switch to side-by-side view

--- a
+++ b/models/CellMovementEncoder.py
@@ -0,0 +1,6 @@
+import torch.nn as nn
+
+class CellMovementEncoder(nn.Module):
+    def __init__(self):
+        super(CellMovementEncoder, self).__init__()
+    # ... Define the architecture and forward pass for CellMovementEncoder as needed