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 |