def setup()

in projects/deep_video_compression/data_module.py [0:0]

• 21 lines of code
• 1 McCabe index (conditional complexity)

    def setup(self, stage: Optional[str] = None) -> None:
        train_transforms = transforms.Compose(
            [transforms.CenterCrop(self.image_size), transforms.ToTensor()]
        )

        val_transforms = transforms.Compose(
            [transforms.CenterCrop(self.image_size), transforms.ToTensor()]
        )

        self.train_dataset = Vimeo90kSeptuplet(
            root=self.data_dir,
            frames_per_group=self.frames_per_group,
            pil_transform=train_transforms,
            as_video=True,
            split="train",
        )

        self.val_dataset = Vimeo90kSeptuplet(
            self.data_dir,
            frames_per_group=self.frames_per_group,
            pil_transform=val_transforms,
            as_video=True,
            split="test",
        )