rivesunder / dca Goto Github PK

This will enable faster training by making sub-classing differentiable cellular automata from torch.nn.Module, making it compatible with multi-gpu traing with torch.nn.DataParallel

An added bonus is that this will create a reusable and flexible model module for using dCA in many other applications besides, including all types of computer vision tasks but also some more imaginative applications.