Spaces:
Sleeping
Sleeping
| viton_hd_train: | |
| dataset: | |
| _target_: media_dataloader.api.EnrichingDataset | |
| datasource: | |
| _target_: media_dataloader.api.LazyHiveDataSource | |
| namespace: ad_metrics | |
| table: vton_public_dataset_viton_hd_train_v2 | |
| partition_filter_predicate_list: ["ds = '2024-10-30'"] | |
| enrichments: | |
| - _target_: media_dataloader.api.media_lookups.ManifoldLookups | |
| lookup_handle_to_media_columns: | |
| image_manifold_path: "image" | |
| cloth_manifold_path: "cloth" | |
| agnostic_mask_manifold_path: "agnostic_mask" | |
| image_densepose_manifold_path: "image_densepose" | |
| cloth_mask_manifold_path: "cloth_mask" | |
| image_parse_v3_manifold_path: "image_parse" | |
| collate_fn: | |
| - _target_: media_dataloader.api.Collate | |
| - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform | |
| image_field: image | |
| blob_field: image | |
| - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform | |
| image_field: cloth | |
| blob_field: cloth | |
| - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform | |
| image_field: agnostic_mask | |
| blob_field: agnostic_mask | |
| - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform | |
| image_field: image_densepose | |
| blob_field: image_densepose | |
| - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform | |
| image_field: cloth_mask | |
| blob_field: cloth_mask | |
| - _target_: torchmultimodal.fb.genai.transforms.hive_transforms.EverstoreImageToPILTransform | |
| image_field: image_parse | |
| blob_field: image_parse | |
| - _target_: leffa.datasets.transform.VtonTransform | |
| height: ${constants.height} | |
| width: ${constants.width} | |
| is_train: true | |
| dataset: viton_hd | |
| aug_garment_ratio: 0.0 | |
| get_garment_from_person_ratio: 0.0 | |
| aug_mask_ratio: 0.0 | |
| dataloader: | |
| _target_: media_dataloader.api.StatefulDataLoader | |
| dataset: ${datasets.viton_hd_train.dataset} | |
| batch_size: ${constants.batch_size} | |
| num_workers: 4 | |
| prefetch_factor: 2 | |
| pin_memory: true | |
| persistent_workers: true | |
| multiprocessing_context: forkserver | |