diff --git a/ppcls/modeling/architectures/ghostnet.py b/ppcls/modeling/architectures/ghostnet.py index 8f9c54b625e547018f617da3e1c886f71776a6fd..038e2f39fd5f77dc93d47790c9edb0447191a6ad 100644 --- a/ppcls/modeling/architectures/ghostnet.py +++ b/ppcls/modeling/architectures/ghostnet.py @@ -263,17 +263,3 @@ def GhostNet_x1_0(): def GhostNet_x1_3(): model = GhostNet(scale=1.3) return model - -if __name__ == "__main__": - image = fluid.data(name='image', shape=[16, 3, 224, 224], dtype='float32') - - model = GhostNet_x1_0() - out = model.net(input=image, class_dim=1000) - test_program = fluid.default_main_program().clone(for_test=True) - - place = fluid.CPUPlace() - exe = fluid.Executor(place) - exe.run(fluid.default_startup_program()) - - #lltotal_flops_params, is_quantize = summary(test_program) - fluid.save(test_program, "ghostnet1_3") \ No newline at end of file