Python save_and_load_benchmarkの例

コード例 #1

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

コード例 #2

ファイル: inception_resnet_v2_benchmark_test.py プロジェクト: TheVinhLuong102/keras

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

コード例 #3

ファイル: nasnet_large_benchmark_test.py プロジェクト: TheVinhLuong102/keras

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

コード例 #4

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

コード例 #5

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

コード例 #6

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

コード例 #7

ファイル: densenet_benchmark_test.py プロジェクト: paolodedios/keras

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

コード例 #8

ファイル: nasnet_large_benchmark_test.py プロジェクト: bhardwajRahul/keras

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

コード例 #9

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