reduce ut time

test=develop
revert-16045-imperative_remove_desc
minqiyang 7 years ago committed by ceci3
parent c90b82a637
commit 94c8ce3f13

@ -136,10 +136,7 @@ class VarBase {
public: public:
virtual ~VarBase() { virtual ~VarBase() {
if (block_ && !persistable_) { // TODO(minqiyang): remove var desc from block desc
block_->RemoveVar(name_);
}
if (var_) { if (var_) {
delete var_; delete var_;
var_ = nullptr; var_ = nullptr;

@ -243,7 +243,7 @@ class TestImperativePtbRnn(unittest.TestCase):
dy_loss = None dy_loss = None
last_hidden = None last_hidden = None
last_cell = None last_cell = None
batch_num = 200 batch_num = 50
for i in range(batch_num): for i in range(batch_num):
x_data = np.arange(12).reshape(4, 3).astype('int64') x_data = np.arange(12).reshape(4, 3).astype('int64')

@ -231,7 +231,7 @@ class TestImperativeResnet(unittest.TestCase):
seed = 90 seed = 90
batch_size = train_parameters["batch_size"] batch_size = train_parameters["batch_size"]
batch_num = 50 batch_num = 20
with fluid.imperative.guard(): with fluid.imperative.guard():
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
fluid.default_main_program().random_seed = seed fluid.default_main_program().random_seed = seed

Loading…
Cancel
Save