search

Bertinus / FLeCS

Flexible and Learnable Cell Simulator
MIT License
8 stars 1 forks source link

Simplified the procedure to get a trainable CellPop object #30

Closed Bertinus closed 1 year ago

Bertinus commented 1 year ago

CellPop, NodeSet and EdgeSet classes now inherit from torch.nn.Module.

I override theirparameters() method, so that the parameters get registered automatically.