def check_success(providers, provider_options, expected_providers,
expected_provider_options):
actual_providers, actual_provider_options = check_and_normalize_provider_args(
providers, provider_options, valid_providers)
self.assertEqual(actual_providers, expected_providers)
self.assertEqual(actual_provider_options,
expected_provider_options)
def __init__(self,
path_or_bytes,
parameters,
sess_options=None,
providers=None,
provider_options=None):
Session.__init__(self)
if sess_options:
self._sess = C.TrainingSession(sess_options)
else:
self._sess = C.TrainingSession()
# providers needs to be passed explicitly as of ORT 1.10
# retain the pre-1.10 behavior by setting to the available providers.
if providers is None:
providers = C.get_available_providers()
providers, provider_options = check_and_normalize_provider_args(
providers, provider_options, C.get_available_providers())
if isinstance(path_or_bytes, str):
config_result = self._sess.load_model(path_or_bytes, parameters,
providers, provider_options)
elif isinstance(path_or_bytes, bytes):
config_result = self._sess.read_bytes(path_or_bytes, parameters,
providers, provider_options)
else:
raise TypeError("Unable to load from type '{0}'".format(
type(path_or_bytes)))
self.loss_scale_input_name = config_result.loss_scale_input_name
self._inputs_meta = self._sess.inputs_meta
self._outputs_meta = self._sess.outputs_meta
def __init__(self,
path_or_bytes,
parameters,
sess_options=None,
providers=None,
provider_options=None):
Session.__init__(self)
if sess_options:
self._sess = C.TrainingSession(sess_options)
else:
self._sess = C.TrainingSession()
providers, provider_options = check_and_normalize_provider_args(
providers, provider_options, C.get_available_providers())
if isinstance(path_or_bytes, str):
config_result = self._sess.load_model(path_or_bytes, parameters,
providers, provider_options)
elif isinstance(path_or_bytes, bytes):
config_result = self._sess.read_bytes(path_or_bytes, parameters,
providers, provider_options)
else:
raise TypeError("Unable to load from type '{0}'".format(
type(path_or_bytes)))
self.loss_scale_input_name = config_result.loss_scale_input_name
self._inputs_meta = self._sess.inputs_meta
self._outputs_meta = self._sess.outputs_meta
def check_failure(providers, provider_options):
with self.assertRaises(ValueError):
check_and_normalize_provider_args(providers, provider_options, valid_providers)
