c6edefdca203a4b58418684295203c3b521cd5bc,opennmt/training.py,Trainer,__init__,#Trainer#Any#Any#Any#,17

Before Change

      devices = tf.config.experimental.list_logical_devices(device_type="GPU")
      if not devices:
        devices = tf.config.experimental.list_logical_devices(device_type="CPU")
      devices = [devices[0].name]
    self._checkpoint = checkpoint
    self._mixed_precision = mixed_precision
    self._model = checkpoint.model
    self._optimizer = checkpoint.optimizer

After Change

    if checkpoint.optimizer is None:
      raise ValueError("No optimizer is defined")
    if not devices:
      devices = misc.get_devices(count=1)  // Train with 1 device by default.
    self._checkpoint = checkpoint
    self._mixed_precision = mixed_precision
    self._model = checkpoint.model

Instances