INNER CODE UNIT · Python

init_wandb

digantamisra98/Mish · PyTorch Benchmarks/train_imagenet.py:154

def init_wandb(entity, project, model):
    wandb.init(entity=entity, project=project)
    wandb.config.update(args)
    wandb.watch(model)


def set_seed(seed):
    torch.manual_seed(seed)
    torch.cuda.manual_seed_all(seed)
    random.seed(seed)
    torch.backends.cudnn.deterministic = True


def get_model(arch, depth):
    assert arch == "resnet"
    model = ResidualNet("ImageNet", depth, 1000)
    print(model)
    return model

View source record →