Refactor train_network.py (and related stuff)

The goal here is to refactor this script and all related implementations, to make them compliant to the current framework

To upload designs, you'll need to enable LFS and have an admin enable hashed storage. More information