INNER CODE UNIT · Python

model

adobe/antialiased-cnns · main.py:216

        model = antialiased_cnns.__dict__[args.arch[:-5]](pretrained=args.pretrained, 
                                                          filter_size=int(args.arch[-1]), 
                                                          _force_nonfinetuned=args.force_nonfinetuned)
    else: # baseline model
        model = models.__dict__[args.arch](pretrained=args.pretrained)

    # instrumentation
    if(args.wandb):
        import wandb
        wandb.init(project='antialiased-cnns')
        wandb.config.update(args)
        wandb.watch(model)

    if args.finetune: # finetune from baseline "aliased" model
        print("=> copying over pretrained weights from [%s]"%args.arch[:-5])
        model_baseline = models.__dict__[args.arch[:-5]](pretrained=True)
        antialiased_cnns.copy_params_buffers(model_baseline, model)

View source record →