(self)
return res
Collect relevant hyperparamters specifying an experiment.
Collect relevant hyperparamters specifying an experiment.
Collect relevant hyperparamters specifying an experiment .
def hparams(self):
        """Collect relevant hyperparamters specifying an experiment."""
        res = {}
        res.update(self.network.hparams)
        res.update(self.train_dataloader.hparams)
        res.update({'device': str(self.device),
                    'dtype': str(self.dtype),
                    'loss': str(self.loss),
                    'weight_decay': self.optimizer.defaults.get('weight_decay', ''),
                    'lr': self.optimizer.defaults.get('lr', '')
                    })

        return res
def hparams ( self ) : res = { } res . update ( self . network . hparams ) res . update ( self . train_dataloader . hparams ) res . update ( { 'device' : str ( self . device ) , 'dtype' : str ( self . dtype ) , 'loss' : str ( self . loss ) , 'weight_decay' : self . optimizer . defaults . get ( 'weight_decay' , '' ) , 'lr' : self . optimizer . defaults . get ( 'lr' , '' ) } ) return res