Exemplo n.º 1
0
def get_internal_checkpoint_dir(seq_length):
    '''
    Get the location of the checkpoint directory for the given sequence length
    '''
    internal_checkpoint_dir = os.path.join(restore_config.get_restore_dir(),
                                           "seq-%d" % seq_length)

    # if it does not exist, make it
    if not os.path.isdir(internal_checkpoint_dir):
        os.makedirs(internal_checkpoint_dir)
    return internal_checkpoint_dir
Exemplo n.º 2
0
def get_internal_checkpoint_dir(seq_length):
    internal_checkpoint_dir = os.path.join(restore_config.get_restore_dir(),
                                           "seq-%d" % seq_length)
    if not os.path.isdir(internal_checkpoint_dir):
        os.makedirs(internal_checkpoint_dir)
    return internal_checkpoint_dir