PyTorch framework for Deep Learning research and development. It focuses on reproducibility, rapid experimentation, and codebase reuse so you can create something new rather than write another regular train loop. Break the cycle - use the Catalyst!
Specify loaders from catalyst dict:
Sequential( (0): Flatten() (1): Linear(in_features=784, out_features=10, bias=True) )