95d6cf194274246ee029416186f60806617d8fe6,engine/training.py,,run,#Any#Any#Any#Any#,24

Before Change

                previous_sub_dir = max([int(name) for name in log_sub_dirs])
            except:
                quit("Any immediate sub directories of the log directory must be integer-numbered")
            if param.starting_iter > 0:
                current_log_sub_dir = str(previous_sub_dir)
            else:
                current_log_sub_dir = str(1+previous_sub_dir)
        else:
            current_log_sub_dir = "0"
        writer = tf.summary.FileWriter(os.path.join(root_dir, "logs", current_log_sub_dir),
                                       sess.graph)

After Change

        log_sub_dirs = [name for name in os.listdir(os.path.join(root_dir, "logs")) if name.isdecimal()]
        if log_sub_dirs and param.starting_iter == 0:
            log_sub_dir = str(max([int(name) for name in log_sub_dirs]) + 1)
        elif log_sub_dirs and param.starting_iter > 0:
            log_sub_dir = str(
                max([int(name) for name in log_sub_dirs if os.path.isdir(os.path.join(root_dir, "logs", name))]))
        else:
            log_sub_dir = "0"
        writer = tf.summary.FileWriter(os.path.join(root_dir, "logs", log_sub_dir),
                                       sess.graph)

Instances