jihunchoi

import os
from typing import List, Optional

import boto3
from hydra.core.object_type import ObjectType
from hydra.plugins.config_source import ConfigResult, ConfigSource
from omegaconf import OmegaConf
from smart_open import open, parse_uri

jihunchoi

def _sequence_mask(sequence_length, max_len=None):
    if max_len is None:
        max_len = sequence_length.data.max()
    batch_size = sequence_length.size(0)
    seq_range = torch.range(0, max_len - 1).long()
    seq_range_expand = seq_range.unsqueeze(0).expand(batch_size, max_len)
    seq_range_expand = Variable(seq_range_expand)
    if sequence_length.is_cuda:
        seq_range_expand = seq_range_expand.cuda()
    seq_length_expand = (sequence_length.unsqueeze(1)

jihunchoi

"""Clean all experiments in a PyCrayon server."""
from pycrayon import CrayonClient


def main():
    client = CrayonClient()
    client.remove_all_experiments()


if __name__ == '__main__':