|
dress_code_train: |
|
dataset: |
|
_target_: media_dataloader.api.EnrichingDataset |
|
datasource: |
|
_target_: media_dataloader.api.LazyHiveDataSource |
|
namespace: ad_metrics |
|
table: vton_public_dataset_dress_code_train_v2 |
|
|
|
partition_filter_predicate_list: ["ds = '2024-09-14'"] |
|
|
|
|
|
|
|
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" |
|
dense_manifold_path: "image_densepose" |
|
label_map_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: image_parse |
|
blob_field: image_parse |
|
- _target_: leffa.datasets.transform.VtonTransform |
|
height: ${constants.height} |
|
width: ${constants.width} |
|
is_train: true |
|
dataset: dress_code |
|
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.dress_code_train.dataset} |
|
batch_size: ${constants.batch_size} |
|
num_workers: 4 |
|
prefetch_factor: 2 |
|
pin_memory: true |
|
persistent_workers: true |
|
multiprocessing_context: forkserver |
|
|
