Open Vosatorp opened 2 months ago
The problem is that 'blade' is not registered in Registry file. The solution is to change the value of backbone to ncsnpp.
If you check in the init() of the ScoreModel class
class ScoreModel(pl.LightningModule):
def __init__(self,
backbone: str = "blade", sde: str = "ouvesde",
lr: float = 1e-4, ema_decay: float = 0.999,
t_eps: float = 3e-2, transform: str = 'none', nolog: bool = False,
num_eval_files: int = 50, loss_type: str = 'mse', data_module_cls = None, **kwargs
):
Just change that value of ncsnpp. Should look like this:
class ScoreModel(pl.LightningModule):
def __init__(self,
backbone: str = "ncsnpp", sde: str = "ouvesde",
lr: float = 1e-4, ema_decay: float = 0.999,
t_eps: float = 3e-2, transform: str = 'none', nolog: bool = False,
num_eval_files: int = 50, loss_type: str = 'mse', data_module_cls = None, **kwargs
):
Worked for me. Should work for you as well.
Best of luck!!
Does anyone know what the problem might be?