def get_cache_path_run(self, args, epoch):
    cache_dir = (args.data.output.path.parent / 'caches')
    cache_dir.mkdir(parents=True, exist_ok=True)
    pid = args.parent_pid
    rank = args.node_rank
    i = args.chunk_num
    name = 'contrastive_model_cache_epoch_{}_{}_{}_{}.pkl'.format(epoch, pid, rank, i)
    path = str((cache_dir / name))
    key_name = 'contrastive_model_cache_epoch_{}_{}_{}_{}.json'.format(epoch, pid, rank, i)
    key_path = str((cache_dir / key_name))
    return (path, key_path)
def get_cache_path_run(self, args, epoch):
    cache_dir = (args.data.output.path.parent / 'caches')
    cache_dir.mkdir(parents=True, exist_ok=True)
    pid = args.parent_pid
    rank = args.node_rank
    i = args.chunk_num
    name = 'contrastive_model_cache_epoch_{}_{}_{}_{}.pkl'.format(epoch, pid, rank, i)
    path = str((cache_dir / name))
    key_name = 'contrastive_model_cache_epoch_{}_{}_{}_{}.json'.format(epoch, pid, rank, i)
    key_path = str((cache_dir / key_name))
    return (path, key_path)