projects/deep_video_compression/_utils.py [122:133]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.num_pframes = num_pframes
        if loss_functions.entropy_fn is None:
            raise ValueError("Must specify entropy_fn for compression stage.")
        self.loss_functions = loss_functions

    def forward(self, image1, image2):
        flow = self.motion_estimator(image1, image2)

        latent, sizes = self.motion_encoder(flow)
        latent, probabilities = self.motion_entropy_bottleneck(latent)
        flow = self.motion_decoder(latent, sizes)
        image2_est = ncF.dense_image_warp(image1, flow.permute(0, 2, 3, 1))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



projects/deep_video_compression/_utils.py [191:202]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self.num_pframes = num_pframes
        if loss_functions.entropy_fn is None:
            raise ValueError("Must specify entropy_fn for compression stage.")
        self.loss_functions = loss_functions

    def forward(self, image1, image2):
        flow = self.motion_estimator(image1, image2)

        latent, sizes = self.motion_encoder(flow)
        latent, probabilities = self.motion_entropy_bottleneck(latent)
        flow = self.motion_decoder(latent, sizes)
        image2_est = ncF.dense_image_warp(image1, flow.permute(0, 2, 3, 1))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -