Added train_torch script containing base_train function, used to train the Chambon18 model architecture.