self.log('val_loss', loss, prog_bar=True, on_epoch=True, on_step=True) return loss def configure_optimizers(self): optimizer = torch.optim.Adam(self.parameters(), lr = 2e-4) return optimizer Optmizer = locate(self.optimizer['optimizer']) return Optmizer(self.parameters(), **self.optim...
framework = framework # Initialize the members def Initialize(self): None def Run(self): None def GetObject(framework): obj = DeffeExploration(framework) return obj The "GetObject" method returns the object of the class. It should take the Deffe framework object as an input to configure the...