diff --git a/demo/models/slimfacenet.py b/demo/models/slimfacenet.py index 33a4deab9340855398c9c9e7d0aa4386a6377030..cb9353c706709caa956b9e605588c101ef2869ad 100644 --- a/demo/models/slimfacenet.py +++ b/demo/models/slimfacenet.py @@ -334,7 +334,7 @@ class SlimFaceNet(): else: pass - one_hot = fluid.one_hot(input=label, depth=out_dim) + one_hot = fluid.layers.one_hot(input=label, depth=out_dim) output = fluid.layers.elementwise_mul( one_hot, phi) + fluid.layers.elementwise_mul( (1.0 - one_hot), cosine) @@ -367,7 +367,8 @@ def SlimFaceNet_C_x0_75(class_dim=None, scale=0.6, arch=None): if __name__ == "__main__": + paddle.enable_static() x = fluid.data(name='x', shape=[-1, 3, 112, 112], dtype='float32') print(x.shape) - model = SlimFaceNet(10000, [1, 3, 3, 1, 1, 0, 0, 1, 0, 1, 1, 0, 5, 5, 3]) + model = SlimFaceNet(10000, arch=[1, 3, 3, 1, 1, 0, 0, 1, 0, 1, 1, 0, 5, 5, 3]) y = model.net(x) diff --git a/demo/slimfacenet/train_eval.py b/demo/slimfacenet/train_eval.py index 77b366b92842445616e022d8c43a377c749fb832..82166305dd1f4f854877281617e59abccefcc12f 100644 --- a/demo/slimfacenet/train_eval.py +++ b/demo/slimfacenet/train_eval.py @@ -29,6 +29,7 @@ from lfw_eval import parse_filelist, evaluation_10_fold from paddleslim import models from paddleslim.quant import quant_post_static +paddle.enable_static() def now(): return time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))