device_of

hybrid_learning.concepts.train_eval.train_eval_funs.device_of(model)[source]

Return the device of the given pytorch model. Distributed models are not supported.