未验证 提交 2ff97018 编写于 作者: C ceci3 提交者: GitHub

modify flops constraint (#859)

上级 7ea07cab
...@@ -16,6 +16,7 @@ from __future__ import absolute_import ...@@ -16,6 +16,7 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import os
import math import math
import paddle import paddle
......
...@@ -165,7 +165,7 @@ for step in range(3): ...@@ -165,7 +165,7 @@ for step in range(3):
train_loader, eval_loader = input_data(images, label) train_loader, eval_loader = input_data(images, label)
current_flops = slim.analysis.flops(train_program) current_flops = slim.analysis.flops(train_program)
if current_flops > 321208544: if current_flops > 6555276:
continue continue
for epoch in range(7): for epoch in range(7):
......
...@@ -164,7 +164,7 @@ for step in range(3): ...@@ -164,7 +164,7 @@ for step in range(3):
train_loader, eval_loader = input_data(images, label) train_loader, eval_loader = input_data(images, label)
current_flops = slim.analysis.flops(train_program) current_flops = slim.analysis.flops(train_program)
if current_flops > 321208544: if current_flops > 6555276:
continue continue
for epoch in range(7): for epoch in range(7):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册