a38ac8958ebf1c61a10c4d5320f1e31d3d0b73dd,trainer.py,Trainer,__init__,#Trainer#,27

Before Change

        if self.use_gpu:
            self.model.cuda()

        print("[*] Number of model parameters: {:,}".format(
            sum([p.data.nelement() for p in self.model.parameters()])))

        // // initialize optimizer and scheduler
        // self.optimizer = optim.SGD(
        //     self.model.parameters(), lr=self.lr, momentum=self.momentum,

After Change

        
        self.config = config

        if config.use_gpu and torch.cuda.is_available():
            self.device = torch.device("cuda")
        else:
            self.device = torch.device("cpu")

        // glimpse network params
        self.patch_size = config.patch_size
        self.glimpse_scale = config.glimpse_scale
        self.num_patches = config.num_patches
        self.loc_hidden = config.loc_hidden

Instances