Add some trace output in Training#train
Fix mistake in d132a23f1b3de58040542e8ce1dd9e8a9b88f0ca

Forgot to actually add the classifier layers to the data structure...