Update trainer.py

change _create_layer to create_layer