Check if model is eval or train

[Rafael_Valle]

That’s easy! Thank you!