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
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
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"])