fix serial number

shanyi15-patch-3
tangwei12 7 years ago
parent 22df4c278c
commit d98480cff5

@ -114,8 +114,8 @@ class CheckpointLoadOp : public framework::OperatorBase {
std::string dir = Attr<std::string>("dir");
std::string serial_num_attr = Attr<std::string>("Serial");
PADDLE_ENFORCE(IsNumber(serial_num_attr),
"Checkpoint Serial must be a number");
VLOG(3) << "CheckpointLoadOp get Attr dir: " << dir;
VLOG(3) << "CheckpointLoadOp get Attr Serial: " << serial_num_attr;
std::string serial_var_name = std::string(SERIAL_VAR);
auto *serial_var = scope.FindVar(serial_var_name);

@ -654,6 +654,9 @@ class DistributeTranspiler:
if os.path.isfile(success_path):
return int(cur_dir)
if os.path.isdir(checkpoint_dir):
return "-1"
current_dir = 0
dirs = os.listdir(checkpoint_dir)
for cur_dir in dirs:

Loading…
Cancel
Save