Switch to unified view

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