Add interfaces to neural nets for pretraining