diff --git a/src/model.py b/src/model.py index 5dfc80de..66b1b48f 100644 --- a/src/model.py +++ b/src/model.py @@ -29,7 +29,7 @@ def __init__(self): no_relu_layers = ['conv5_5_CPM_L1', 'conv5_5_CPM_L2', 'Mconv7_stage2_L1',\ 'Mconv7_stage2_L2', 'Mconv7_stage3_L1', 'Mconv7_stage3_L2',\ 'Mconv7_stage4_L1', 'Mconv7_stage4_L2', 'Mconv7_stage5_L1',\ - 'Mconv7_stage5_L2', 'Mconv7_stage6_L1', 'Mconv7_stage6_L1'] + 'Mconv7_stage5_L2', 'Mconv7_stage6_L1', 'Mconv7_stage6_L2'] blocks = {} block0 = OrderedDict([ ('conv1_1', [3, 64, 3, 1, 1]),