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

modify flops constraint (#859)

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