Exemplo n.º 1
0
    def benchmark_save_and_load_vgg19(self):
        app = tf.keras.applications.VGG19
        save_result, load_result = (
            saved_model_benchmark_util.save_and_load_benchmark(app))

        self.report_benchmark(iters=save_result['iters'],
                              wall_time=save_result['wall_time'],
                              name=save_result['name'])

        self.report_benchmark(iters=load_result['iters'],
                              wall_time=load_result['wall_time'],
                              name=load_result['name'])
    def benchmark_save_and_load_inception_resnet_v2(self):
        app = tf.keras.applications.InceptionResNetV2
        save_result, load_result = (
            saved_model_benchmark_util.save_and_load_benchmark(app))

        self.report_benchmark(iters=save_result['iters'],
                              wall_time=save_result['wall_time'],
                              name=save_result['name'])

        self.report_benchmark(iters=load_result['iters'],
                              wall_time=load_result['wall_time'],
                              name=load_result['name'])
Exemplo n.º 3
0
    def benchmark_save_and_load_efficient_net_b7(self):
        app = tf.keras.applications.EfficientNetB7
        save_result, load_result = (
            saved_model_benchmark_util.save_and_load_benchmark(app))

        self.report_benchmark(iters=save_result['iters'],
                              wall_time=save_result['wall_time'],
                              name=save_result['name'])

        self.report_benchmark(iters=load_result['iters'],
                              wall_time=load_result['wall_time'],
                              name=load_result['name'])
    def benchmark_save_and_load_nasnet_large(self):
        app = tf.keras.applications.NASNetLarge
        save_result, load_result = (
            saved_model_benchmark_util.save_and_load_benchmark(app))

        self.report_benchmark(iters=save_result['iters'],
                              wall_time=save_result['wall_time'],
                              name=save_result['name'])

        self.report_benchmark(iters=load_result['iters'],
                              wall_time=load_result['wall_time'],
                              name=load_result['name'])
Exemplo n.º 5
0
    def benchmark_save_and_load_densenet_201(self):
        app = tf.keras.applications.DenseNet201
        save_result, load_result = (
            saved_model_benchmark_util.save_and_load_benchmark(app))

        self.report_benchmark(iters=save_result['iters'],
                              wall_time=save_result['wall_time'],
                              name=save_result['name'])

        self.report_benchmark(iters=load_result['iters'],
                              wall_time=load_result['wall_time'],
                              name=load_result['name'])