添加@tf.function装饰
我们分析入口文件official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py，可以看到官方脚本已经默认添加了@tf.function装饰，因而这个迁移点我们直接迁移完成。
1 2 3 | flags.DEFINE_boolean(name='use_tf_function', default=True, help='Wrap the train and test step inside a ' 'tf.function.') |
父主题： 手工迁移与训练