Generalisable Agents for Neural Network Optimisation