Exemplos de parse_dtype_info em Python

Exemplo n.º 1

    def parse_args(self, args=None, namespace=None):
        args = super(ResnetArgParser, self).parse_args(args=args,
                                                       namespace=namespace)

        # handle coupling between dtype and loss_scale
        parsers.parse_dtype_info(args)

        return args

Exemplo n.º 2

Arquivo: resnet_run_loop.py Projeto: smajida/models

  def parse_args(self, args=None, namespace=None):
    args = super(ResnetArgParser, self).parse_args(
        args=args, namespace=namespace)

    # handle coupling between dtype and loss_scale
    parsers.parse_dtype_info(args)

    return args

Exemplo n.º 3

Arquivo: parsers_test.py Projeto: cybermaster/reference

  def test_parse_dtype_info(self):
    parser = TestParser()
    for dtype_str, tf_dtype, loss_scale in [["fp16", tf.float16, 128],
                                            ["fp32", tf.float32, 1]]:
      args = parser.parse_args(["--dtype", dtype_str])
      parsers.parse_dtype_info(args)

      assert args.dtype == tf_dtype
      assert args.loss_scale == loss_scale

      args = parser.parse_args(["--dtype", dtype_str, "--loss_scale", "5"])
      parsers.parse_dtype_info(args)

      assert args.loss_scale == 5

    with self.assertRaises(SystemExit):
      parser.parse_args(["--dtype", "int8"])

Exemplo n.º 4

  def test_parse_dtype_info(self):
    parser = TestParser()
    for dtype_str, tf_dtype, loss_scale in [["fp16", tf.float16, 128],
                                            ["fp32", tf.float32, 1]]:
      args = parser.parse_args(["--dtype", dtype_str])
      parsers.parse_dtype_info(args)

      assert args.dtype == tf_dtype
      assert args.loss_scale == loss_scale

      args = parser.parse_args(["--dtype", dtype_str, "--loss_scale", "5"])
      parsers.parse_dtype_info(args)

      assert args.loss_scale == 5

    with self.assertRaises(SystemExit):
      parser.parse_args(["--dtype", "int8"])