Method __init__

model/base.py:145–147

(self, args, device)

class SelfForcingModel(BaseModel):
    def __init__(self, args, device):
        super().__init__(args, device)
        self.denoising_loss_func = get_denoising_loss(getattr(args, "denoising_loss_type", "flow"))()

    def _run_generator(
        self,