adapted version of the repository https://github.com/OHOHA3/SCI-FI-DICE-Neural_Networks.git + base scheduler