How to access the details of the model.

[arkpku]

This is similar to #279.

so after training a model, what I got is the metrics and whatever tensorboard has to offer. How to access the details of the model to ckeck whether it works as intended, e.g. how to check the hidden states / the embeddings ?

thanks.