Train On TrainingSession
train_on_session(sess, device = NULL)
Session instance on which to perform the operation
instance of DeviceDescriptor