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