diff --git a/CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.json b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.json new file mode 100644 index 0000000000000000000000000000000000000000..753b5a253e63c55059b049154a33eca3c2f2a9e8 --- /dev/null +++ b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.json @@ -0,0 +1,156 @@ +{ + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "base_generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 4, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 20 + }, + "output": "data/lils/dsprites_large_6_20_16_16.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 16 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 16 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "composition": { + "action_factors": [ + "foreground_color", + "background_color", + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "uniform", + "composition_op": "interp", + "ignored_factors": [] + }, + "n_shapes_sweep": [ + 3 + ], + "num_reruns": 1, + "rerun_offset": 0, + "runs": [ + { + "held_out_shape": "polygon_3", + "n_shapes": 3, + "rerun_index": 0, + "run_index": 0, + "run_seed": 2548879074, + "run_shapes": [ + "polygon_3", + "polygon_5", + "star_5_r04" + ], + "z_dim": 10 + } + ], + "shape_pool": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "sprite_right_side_threshold": 0.5, + "sprites_large_cache": "data/lils/dsprites_large_6_20_16_16.pt", + "val_fraction": 0.0, + "z_dims": [ + 10 + ] +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.yaml b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.yaml new file mode 100644 index 0000000000000000000000000000000000000000..57d4bc53ea9947833d86e298331faa9acc5d37b4 --- /dev/null +++ b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.yaml @@ -0,0 +1,131 @@ +sprites_large_cache: data/lils/dsprites_large_6_20_16_16.pt +available_shapes: +- polygon_3 +- polygon_4 +- polygon_5 +- star_3_r02 +- star_4_r04 +- star_5_r04 +- star_6_r04 +- ellipse_050 +- heart +- flower_2_w03 +- flower_3_w03 +- flower_4_w03 +- flower_5_w03 +- flower_6_w03 +- rectangle_03 +- cross_065 +- crescent_025 +- drop_15 +- closed_hollow_polygon_3 +- closed_hollow_polygon_4 +- closed_hollow_polygon_5 +- hollow_polygon_3 +- hollow_polygon_4 +- hollow_polygon_5 +- closed_hollow_ellipse_05 +shape_pool: +- polygon_3 +- polygon_4 +- polygon_5 +- star_3_r02 +- star_4_r04 +- star_5_r04 +- star_6_r04 +- ellipse_050 +- heart +- flower_2_w03 +- flower_3_w03 +- flower_4_w03 +- flower_5_w03 +- flower_6_w03 +- rectangle_03 +- cross_065 +- crescent_025 +- drop_15 +- closed_hollow_polygon_3 +- closed_hollow_polygon_4 +- closed_hollow_polygon_5 +- hollow_polygon_3 +- hollow_polygon_4 +- hollow_polygon_5 +- closed_hollow_ellipse_05 +num_reruns: 1 +rerun_offset: 0 +sprite_right_side_threshold: 0.5 +val_fraction: 0.0 +n_shapes_sweep: +- 3 +z_dims: +- 10 +base_generation_config: + seed: 42 + image_size: 64 + num_values: 4 + output_channels: 1 + output: data/lils/dsprites_large_6_20_16_16.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 20 + positions_x: + min: 0.2 + max: 0.8 + num: 16 + positions_y: + min: 0.2 + max: 0.8 + num: 16 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +composition: + composition_op: interp + command_sampler: uniform + action_factors: + - foreground_color + - background_color + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: [] +runs: +- run_index: 0 + rerun_index: 0 + run_seed: 2548879074 + n_shapes: 3 + z_dim: 10 + held_out_shape: polygon_3 + run_shapes: + - polygon_3 + - polygon_5 + - star_5_r04 diff --git a/CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..03646eb59f030385a72584e5b1083d69652cf040 --- /dev/null +++ b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,142 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 4, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 20 + }, + "output": "data/lils/dsprites_large_6_20_16_16.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 16 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 16 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/lils/dsprites_large_6_20_16_16.pt" + }, + "composition": { + "action_factors": [ + "foreground_color", + "background_color", + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "uniform", + "composition_op": "interp", + "ignored_factors": [] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "polygon_3" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "polygon_3", + "polygon_5", + "star_5_r04" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "polygon_3", + "n_shapes": 3, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_03shapes_zdim_010", + "run_seed": 2548879074, + "run_shapes": [ + "polygon_3", + "polygon_5", + "star_5_r04" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 9.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 1, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 4, + "seed": 2548879074 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..985b72d96f33e912d4de7daae0626adfdaf2486d --- /dev/null +++ b/CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,118 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_03shapes_zdim_010 +run_seed: 2548879074 +z_dim: 10 +n_shapes: 3 +held_out_shape: polygon_3 +run_shapes: +- polygon_3 +- polygon_5 +- star_5_r04 +cache: + path: data/lils/dsprites_large_6_20_16_16.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 4 + output_channels: 1 + output: data/lils/dsprites_large_6_20_16_16.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 20 + positions_x: + min: 0.2 + max: 0.8 + num: 16 + positions_y: + min: 0.2 + max: 0.8 + num: 16 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - polygon_3 + - polygon_5 + - star_5_r04 + sprite_restricted_shapes: + - polygon_3 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: interp + command_sampler: uniform + action_factors: + - foreground_color + - background_color + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: [] +training: + epochs: 1 + batch_size: 128 + num_workers: 4 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 2548879074 + base_seed: 42 + bbox_loss_inside_weight: 9.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoint_summary.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoint_summary.json new file mode 100644 index 0000000000000000000000000000000000000000..a4f657ba1949d675ef388372b54a94c4d894d8ce --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoint_summary.json @@ -0,0 +1,10 @@ +{ + "checkpoints": { + "best_in_domain": { + "monitor": "val_in_domain_bpd", + "path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt", + "score": 0.04238998889923096 + } + }, + "default_checkpoint": "best_in_domain" +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..86811460b8b06092e68436bcb079f5e22cefdf67 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0ca8d9925c4c94bf295097f74c737103299dcd8a910a4bd8fa8161107b12a46 +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..6614cb1dfa8fa083e3fc7d2822d12f918faeb246 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/dataset_loader_config.json @@ -0,0 +1,212 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32_mmap", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 614400, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 737280, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 2548879074, + "sprite_restricted_shapes": [ + "polygon_3" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "polygon_3", + "polygon_5", + "star_5_r04" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 614400, + "ood_validation_pool_size": 737280, + "train_anchor_size": 614399, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/events.out.tfevents.1775565498.node04.1267668.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/events.out.tfevents.1775565498.node04.1267668.0 new file mode 100644 index 0000000000000000000000000000000000000000..4254df8cfdc0d6089e070385fe18037b6a12eea9 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/events.out.tfevents.1775565498.node04.1267668.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d18f0a7b81e8594fb37720ca4d22362b4aea6176da23a107c2ccc8f86034804c +size 62058 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..1e62d81750829f032a88f42fb99d0de1a41293fc --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/dataset_loader_config.json @@ -0,0 +1,212 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 614400, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 737280, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 2548879074, + "sprite_restricted_shapes": [ + "polygon_3" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "polygon_3", + "polygon_5", + "star_5_r04" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 614400, + "ood_validation_pool_size": 737280, + "train_anchor_size": 614399, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/events.out.tfevents.1775565909.node05.1313440.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/events.out.tfevents.1775565909.node05.1313440.0 new file mode 100644 index 0000000000000000000000000000000000000000..16929b3090c2660465ecda0fd8d7b4dab08f6645 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/events.out.tfevents.1775565909.node05.1313440.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d4c433d7cbc3f1154774af5c022ac73b58c9e572f53d73ce2e98dd596e50725c +size 5466 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoint_summary.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoint_summary.json new file mode 100644 index 0000000000000000000000000000000000000000..6d33d570341df54cbe311a8574ec16146b11ea20 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoint_summary.json @@ -0,0 +1,10 @@ +{ + "checkpoints": { + "best_in_domain": { + "monitor": "val_in_domain_bpd", + "path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt", + "score": 0.006012121215462685 + } + }, + "default_checkpoint": "best_in_domain" +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..d483c2891a3662a8cf7d167c31a3ac7dbefe4404 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:726558b91b4007537058f3b4710b1b77f2a7088098465cf3d0271ef34c909bab +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..1e62d81750829f032a88f42fb99d0de1a41293fc --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json @@ -0,0 +1,212 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 614400, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 737280, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 2548879074, + "sprite_restricted_shapes": [ + "polygon_3" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "polygon_3", + "polygon_5", + "star_5_r04" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 614400, + "ood_validation_pool_size": 737280, + "train_anchor_size": 614399, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775566849.node04.1274707.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775566849.node04.1274707.0 new file mode 100644 index 0000000000000000000000000000000000000000..738175173179bed72135059528d4c8d3fa256242 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775566849.node04.1274707.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56aeb8f35f22f859c36ed0b18a37efd9a4199c5384d33caf1996d2eedcc61c7c +size 693700 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..78174b888f98cd8e86d0231d562d401de3f169d4 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91df25f9a83df4fd5719e84e782c5fc44e1814c0a32a01974f54614e974c97be +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..4599e85be28d9ddc8cab610b760d433522d2af42 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/dataset_loader_config.json @@ -0,0 +1,215 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 1351680, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 1474560, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 3003674970, + "sprite_restricted_shapes": [ + "star_3_r02" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "star_3_r02", + "star_6_r04", + "heart", + "cross_065", + "crescent_025", + "closed_hollow_polygon_3" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 1351680, + "ood_validation_pool_size": 1474560, + "train_anchor_size": 1351679, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/events.out.tfevents.1775566847.node05.1316752.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/events.out.tfevents.1775566847.node05.1316752.0 new file mode 100644 index 0000000000000000000000000000000000000000..a433e5adb2168a34edb362107f419f510365035c --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/events.out.tfevents.1775566847.node05.1316752.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:feed6a4c297cb46b2d591ed70fd109e973438bf21d14905ea073dff3d08a751e +size 411229 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..abaef3b98e6ac79907180deccc1604d1b67dddc8 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,149 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt" + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "composition_op": "fixint", + "ignored_factors": [ + "foreground_color", + "background_color" + ] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "star_3_r02" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "star_3_r02", + "star_6_r04", + "heart", + "cross_065", + "crescent_025", + "closed_hollow_polygon_3" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "star_3_r02", + "n_shapes": 6, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_06shapes_zdim_010", + "run_seed": 3003674970, + "run_shapes": [ + "star_3_r02", + "star_6_r04", + "heart", + "cross_065", + "crescent_025", + "closed_hollow_polygon_3" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 1.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 100, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 8, + "seed": 3003674970 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..38db23f4a39974d2be01921d52d80ff4b09a5d7f --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,124 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_06shapes_zdim_010 +run_seed: 3003674970 +z_dim: 10 +n_shapes: 6 +held_out_shape: star_3_r02 +run_shapes: +- star_3_r02 +- star_6_r04 +- heart +- cross_065 +- crescent_025 +- closed_hollow_polygon_3 +cache: + path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 2 + output_channels: 1 + output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 40 + positions_x: + min: 0.2 + max: 0.8 + num: 32 + positions_y: + min: 0.2 + max: 0.8 + num: 32 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - star_3_r02 + - star_6_r04 + - heart + - cross_065 + - crescent_025 + - closed_hollow_polygon_3 + sprite_restricted_shapes: + - star_3_r02 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: fixint + command_sampler: lils + action_factors: + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: + - foreground_color + - background_color +training: + epochs: 100 + batch_size: 128 + num_workers: 8 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 3003674970 + base_seed: 42 + bbox_loss_inside_weight: 1.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..c4c1218773e3d4b2c50971cab2a831f4c8f87bcb --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:845924c16efa05d8df19bb498d1e83ed9a0f015d889d277a8642d54e34088b78 +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..2c1e0a90eb494f1e55c58d7800347582a7e06c3d --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/dataset_loader_config.json @@ -0,0 +1,218 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 2088960, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 2211840, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 4083989091, + "sprite_restricted_shapes": [ + "polygon_4" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "polygon_4", + "polygon_5", + "star_5_r04", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "drop_15", + "hollow_polygon_3", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 2088960, + "ood_validation_pool_size": 2211840, + "train_anchor_size": 2088959, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/events.out.tfevents.1775566862.node06.2124470.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/events.out.tfevents.1775566862.node06.2124470.0 new file mode 100644 index 0000000000000000000000000000000000000000..fc181113398dde7c6ba40ed83b4af9ae7aec54f5 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/events.out.tfevents.1775566862.node06.2124470.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7a69014b68191fcce0009f3c8fb3dbd6136e3f43a4ca7b4a251304b04faab2f0 +size 292269 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..03f474349663810bef48e4bd9126a4a86f501d69 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,155 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt" + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "composition_op": "fixint", + "ignored_factors": [ + "foreground_color", + "background_color" + ] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "polygon_4" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "polygon_4", + "polygon_5", + "star_5_r04", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "drop_15", + "hollow_polygon_3", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "polygon_4", + "n_shapes": 9, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_09shapes_zdim_010", + "run_seed": 4083989091, + "run_shapes": [ + "polygon_4", + "polygon_5", + "star_5_r04", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "drop_15", + "hollow_polygon_3", + "closed_hollow_ellipse_05" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 1.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 100, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 8, + "seed": 4083989091 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..2ab090ee4edd0d76e92c1e885e485bb78c881ad8 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,130 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_09shapes_zdim_010 +run_seed: 4083989091 +z_dim: 10 +n_shapes: 9 +held_out_shape: polygon_4 +run_shapes: +- polygon_4 +- polygon_5 +- star_5_r04 +- flower_3_w03 +- flower_4_w03 +- flower_5_w03 +- drop_15 +- hollow_polygon_3 +- closed_hollow_ellipse_05 +cache: + path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 2 + output_channels: 1 + output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 40 + positions_x: + min: 0.2 + max: 0.8 + num: 32 + positions_y: + min: 0.2 + max: 0.8 + num: 32 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - polygon_4 + - polygon_5 + - star_5_r04 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - drop_15 + - hollow_polygon_3 + - closed_hollow_ellipse_05 + sprite_restricted_shapes: + - polygon_4 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: fixint + command_sampler: lils + action_factors: + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: + - foreground_color + - background_color +training: + epochs: 100 + batch_size: 128 + num_workers: 8 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 4083989091 + base_seed: 42 + bbox_loss_inside_weight: 1.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..cdb856eca38be9943828cef480eb7ce81e153731 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3e24dffeb427c7e9eb7da3993697f22c268420e1c2594fd8d69b2f28b088c08d +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..5d77f02234f83dfd1f30cee70bbab96632b00154 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/dataset_loader_config.json @@ -0,0 +1,221 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 2826240, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 2949120, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 941690707, + "sprite_restricted_shapes": [ + "closed_hollow_polygon_5" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "closed_hollow_polygon_5", + "polygon_3", + "polygon_4", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_6_w03", + "cross_065", + "drop_15", + "closed_hollow_polygon_3" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 2826240, + "ood_validation_pool_size": 2949120, + "train_anchor_size": 2826239, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/events.out.tfevents.1775566873.node06.2124623.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/events.out.tfevents.1775566873.node06.2124623.0 new file mode 100644 index 0000000000000000000000000000000000000000..b9b2c2335be9c77605353cf164c80497ee1e5d6c --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/events.out.tfevents.1775566873.node06.2124623.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:676217cc38ca47b224add46fcf6b0c5e258bce43d3cd114413e0a9561de9f5bc +size 217797 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..79dc037fa307afc5ffc00f69367f07fabc893a35 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,161 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt" + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "composition_op": "fixint", + "ignored_factors": [ + "foreground_color", + "background_color" + ] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "closed_hollow_polygon_5" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "closed_hollow_polygon_5", + "polygon_3", + "polygon_4", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_6_w03", + "cross_065", + "drop_15", + "closed_hollow_polygon_3" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "closed_hollow_polygon_5", + "n_shapes": 12, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_12shapes_zdim_010", + "run_seed": 941690707, + "run_shapes": [ + "closed_hollow_polygon_5", + "polygon_3", + "polygon_4", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_6_w03", + "cross_065", + "drop_15", + "closed_hollow_polygon_3" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 1.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 100, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 8, + "seed": 941690707 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..4ae957134ce3bf279a34597a6bf4552aa80435f2 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,136 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_12shapes_zdim_010 +run_seed: 941690707 +z_dim: 10 +n_shapes: 12 +held_out_shape: closed_hollow_polygon_5 +run_shapes: +- closed_hollow_polygon_5 +- polygon_3 +- polygon_4 +- ellipse_050 +- heart +- flower_2_w03 +- flower_3_w03 +- flower_4_w03 +- flower_6_w03 +- cross_065 +- drop_15 +- closed_hollow_polygon_3 +cache: + path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 2 + output_channels: 1 + output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 40 + positions_x: + min: 0.2 + max: 0.8 + num: 32 + positions_y: + min: 0.2 + max: 0.8 + num: 32 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - closed_hollow_polygon_5 + - polygon_3 + - polygon_4 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_6_w03 + - cross_065 + - drop_15 + - closed_hollow_polygon_3 + sprite_restricted_shapes: + - closed_hollow_polygon_5 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: fixint + command_sampler: lils + action_factors: + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: + - foreground_color + - background_color +training: + epochs: 100 + batch_size: 128 + num_workers: 8 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 941690707 + base_seed: 42 + bbox_loss_inside_weight: 1.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..df2abfa00ca65096a5efaf415d58e87e20a16536 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c64772ef2b3840d306ee4d58794be2d39196f3ae9e20cf50a887739468bd542c +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..1e885d4a979dcff3a13f0ef64424ee7645ebbc09 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/dataset_loader_config.json @@ -0,0 +1,224 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 3563520, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 3686400, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 779436189, + "sprite_restricted_shapes": [ + "rectangle_03" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "rectangle_03", + "polygon_3", + "polygon_4", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_5_w03", + "flower_6_w03", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 3563520, + "ood_validation_pool_size": 3686400, + "train_anchor_size": 3563519, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/events.out.tfevents.1775566873.node06.2124788.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/events.out.tfevents.1775566873.node06.2124788.0 new file mode 100644 index 0000000000000000000000000000000000000000..0e24c1db7c3252b3f2b47f21bf22f1717af855ea --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/events.out.tfevents.1775566873.node06.2124788.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ecadb835c91765004915392495fd6736b6af7ff6deeb91a597109ae80ece8472 +size 202001 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..20d28698099dad6fc9c2db22504e603d01770d48 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,167 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt" + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "composition_op": "fixint", + "ignored_factors": [ + "foreground_color", + "background_color" + ] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "rectangle_03" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "rectangle_03", + "polygon_3", + "polygon_4", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_5_w03", + "flower_6_w03", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "rectangle_03", + "n_shapes": 15, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_15shapes_zdim_010", + "run_seed": 779436189, + "run_shapes": [ + "rectangle_03", + "polygon_3", + "polygon_4", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_5_w03", + "flower_6_w03", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 1.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 100, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 8, + "seed": 779436189 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..57e2b123f1f35de6fcc313d241103a26e1508972 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,142 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_15shapes_zdim_010 +run_seed: 779436189 +z_dim: 10 +n_shapes: 15 +held_out_shape: rectangle_03 +run_shapes: +- rectangle_03 +- polygon_3 +- polygon_4 +- star_6_r04 +- ellipse_050 +- heart +- flower_2_w03 +- flower_5_w03 +- flower_6_w03 +- drop_15 +- closed_hollow_polygon_3 +- closed_hollow_polygon_4 +- hollow_polygon_4 +- hollow_polygon_5 +- closed_hollow_ellipse_05 +cache: + path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 2 + output_channels: 1 + output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 40 + positions_x: + min: 0.2 + max: 0.8 + num: 32 + positions_y: + min: 0.2 + max: 0.8 + num: 32 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - rectangle_03 + - polygon_3 + - polygon_4 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_5_w03 + - flower_6_w03 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + sprite_restricted_shapes: + - rectangle_03 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: fixint + command_sampler: lils + action_factors: + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: + - foreground_color + - background_color +training: + epochs: 100 + batch_size: 128 + num_workers: 8 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 779436189 + base_seed: 42 + bbox_loss_inside_weight: 1.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..faf82d20ac97584e9f059c08bc3fec4f6e283357 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4ff58ba8a472875933ff087481084f3cc84c97d1a6b03e4bb6452a0be0fd2732 +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..0ebfaf98d7021e0398b8488f70b7d6e7ce885971 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/dataset_loader_config.json @@ -0,0 +1,229 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 4792320, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 4915200, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 178782122, + "sprite_restricted_shapes": [ + "flower_3_w03" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "flower_3_w03", + "polygon_3", + "polygon_4", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "flower_2_w03", + "flower_4_w03", + "flower_5_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 4792320, + "ood_validation_pool_size": 4915200, + "train_anchor_size": 4792319, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/events.out.tfevents.1775566879.node07.1629786.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/events.out.tfevents.1775566879.node07.1629786.0 new file mode 100644 index 0000000000000000000000000000000000000000..d4e72200de8febca8b90f8c1779c5e9777c818db --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/events.out.tfevents.1775566879.node07.1629786.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2e3086914f2fb894a9ef355038c4e6d7d7145ee143da08b75e2c482dec5d771b +size 142002 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..744a1ba83ba5dcf8a9bfe8bc8a88ff80b8427752 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,177 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt" + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "composition_op": "fixint", + "ignored_factors": [ + "foreground_color", + "background_color" + ] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "flower_3_w03" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "flower_3_w03", + "polygon_3", + "polygon_4", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "flower_2_w03", + "flower_4_w03", + "flower_5_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "flower_3_w03", + "n_shapes": 20, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_20shapes_zdim_010", + "run_seed": 178782122, + "run_shapes": [ + "flower_3_w03", + "polygon_3", + "polygon_4", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "flower_2_w03", + "flower_4_w03", + "flower_5_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 1.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 100, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 8, + "seed": 178782122 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..623596b40c803ae31ed87ee45b835fbb7d3611ee --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,152 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_20shapes_zdim_010 +run_seed: 178782122 +z_dim: 10 +n_shapes: 20 +held_out_shape: flower_3_w03 +run_shapes: +- flower_3_w03 +- polygon_3 +- polygon_4 +- star_3_r02 +- star_4_r04 +- star_5_r04 +- star_6_r04 +- ellipse_050 +- flower_2_w03 +- flower_4_w03 +- flower_5_w03 +- rectangle_03 +- cross_065 +- crescent_025 +- drop_15 +- closed_hollow_polygon_3 +- hollow_polygon_3 +- hollow_polygon_4 +- hollow_polygon_5 +- closed_hollow_ellipse_05 +cache: + path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 2 + output_channels: 1 + output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 40 + positions_x: + min: 0.2 + max: 0.8 + num: 32 + positions_y: + min: 0.2 + max: 0.8 + num: 32 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - flower_3_w03 + - polygon_3 + - polygon_4 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - flower_2_w03 + - flower_4_w03 + - flower_5_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + sprite_restricted_shapes: + - flower_3_w03 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: fixint + command_sampler: lils + action_factors: + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: + - foreground_color + - background_color +training: + epochs: 100 + batch_size: 128 + num_workers: 8 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 178782122 + base_seed: 42 + bbox_loss_inside_weight: 1.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..d592d446badb38ba8cfd04c707712f1262692fb5 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1bc4d211bf2f3927bed98ec191854610d382b74526d80a77620b61c6455a064a +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..e3556e422e9f91ef7d338c76b492fa825b7072a2 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/dataset_loader_config.json @@ -0,0 +1,234 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 6021120, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 6144000, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 1192396634, + "sprite_restricted_shapes": [ + "ellipse_050" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "ellipse_050", + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 6021120, + "ood_validation_pool_size": 6144000, + "train_anchor_size": 6021119, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/events.out.tfevents.1775566899.node07.1629937.0 b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/events.out.tfevents.1775566899.node07.1629937.0 new file mode 100644 index 0000000000000000000000000000000000000000..4205d2b3ca4675784e6818889deb03d6ec81fccc --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/events.out.tfevents.1775566899.node07.1629937.0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ca1e3edceef3c389821126aac829eea9d5b81ec42679cdcf6a71da58781ccc1a +size 131623 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/sweep_run_config.json b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/sweep_run_config.json new file mode 100644 index 0000000000000000000000000000000000000000..7f05961d30a64416615e2bfb35c46a6bda0b0f09 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/sweep_run_config.json @@ -0,0 +1,187 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt" + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "composition_op": "fixint", + "ignored_factors": [ + "foreground_color", + "background_color" + ] + }, + "dataloader": { + "sprite_restricted_shapes": [ + "ellipse_050" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "ellipse_050", + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "val_fraction": 0.0 + }, + "held_out_shape": "ellipse_050", + "n_shapes": 25, + "rerun_index": 0, + "run_index": 0, + "run_name": "run_000_rerun_00_25shapes_zdim_010", + "run_seed": 1192396634, + "run_shapes": [ + "ellipse_050", + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "training": { + "base_seed": 42, + "batch_size": 128, + "bbox_loss_inside_weight": 1.0, + "bbox_loss_outside_weight": 1.0, + "epochs": 100, + "lr": 0.0001, + "num_filters": 64, + "num_values": 2, + "num_workers": 8, + "seed": 1192396634 + }, + "z_dim": 10 +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/sweep_run_config.yaml b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/sweep_run_config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..ae7b58a29aabac32586caa1a3f51c7cb20ff5c19 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/sweep_run_config.yaml @@ -0,0 +1,162 @@ +run_index: 0 +rerun_index: 0 +run_name: run_000_rerun_00_25shapes_zdim_010 +run_seed: 1192396634 +z_dim: 10 +n_shapes: 25 +held_out_shape: ellipse_050 +run_shapes: +- ellipse_050 +- polygon_3 +- polygon_4 +- polygon_5 +- star_3_r02 +- star_4_r04 +- star_5_r04 +- star_6_r04 +- heart +- flower_2_w03 +- flower_3_w03 +- flower_4_w03 +- flower_5_w03 +- flower_6_w03 +- rectangle_03 +- cross_065 +- crescent_025 +- drop_15 +- closed_hollow_polygon_3 +- closed_hollow_polygon_4 +- closed_hollow_polygon_5 +- hollow_polygon_3 +- hollow_polygon_4 +- hollow_polygon_5 +- closed_hollow_ellipse_05 +cache: + path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + available_shapes: + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - ellipse_050 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + generation_config: + seed: 42 + image_size: 64 + num_values: 2 + output_channels: 1 + output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt + scale_space: renderer + position_space: renderer + shape_pool: all + candidate_fg_colors: + - - 255 + - 255 + - 255 + candidate_bg_colors: + - - 0 + - 0 + - 0 + scales: + - 0.2 + - 0.24 + - 0.28 + - 0.32 + - 0.36 + - 0.4 + orientations: + min: 0.0 + max: 6.28318531 + num: 40 + positions_x: + min: 0.2 + max: 0.8 + num: 32 + positions_y: + min: 0.2 + max: 0.8 + num: 32 + train: + num_shapes: 25 + num_fg_colors: 1 + num_bg_colors: 1 + ood: + num_shapes: 0 + shapes_from: unseen + num_fg_colors: 0 + fg_colors_from: unseen + num_bg_colors: 0 + bg_colors_from: unseen +dataloader: + sprite_shapes: + - ellipse_050 + - polygon_3 + - polygon_4 + - polygon_5 + - star_3_r02 + - star_4_r04 + - star_5_r04 + - star_6_r04 + - heart + - flower_2_w03 + - flower_3_w03 + - flower_4_w03 + - flower_5_w03 + - flower_6_w03 + - rectangle_03 + - cross_065 + - crescent_025 + - drop_15 + - closed_hollow_polygon_3 + - closed_hollow_polygon_4 + - closed_hollow_polygon_5 + - hollow_polygon_3 + - hollow_polygon_4 + - hollow_polygon_5 + - closed_hollow_ellipse_05 + sprite_restricted_shapes: + - ellipse_050 + sprite_right_side_threshold: 0.5 + val_fraction: 0.0 +composition: + composition_op: fixint + command_sampler: lils + action_factors: + - shape + - scale + - orientation + - position_x + - position_y + ignored_factors: + - foreground_color + - background_color +training: + epochs: 100 + batch_size: 128 + num_workers: 8 + num_values: 2 + num_filters: 64 + lr: 0.0001 + seed: 1192396634 + base_seed: 42 + bbox_loss_inside_weight: 1.0 + bbox_loss_outside_weight: 1.0 diff --git a/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..8eb7e4b6ae0a9cdbedc984a4b197593706a87b2b --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6a7da9f5b7ce115818eb64f3c50836b5eca1f013b2751a442c4f26ada2ed019 +size 12959138 diff --git a/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json new file mode 100644 index 0000000000000000000000000000000000000000..31282412b0e636604d67373ae5897d00317705d5 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json @@ -0,0 +1,212 @@ +{ + "cache": { + "available_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "generation_config": { + "candidate_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "candidate_fg_colors": [ + [ + 255, + 255, + 255 + ] + ], + "image_size": 64, + "num_values": 2, + "ood": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "num_bg_colors": 0, + "num_fg_colors": 0, + "num_shapes": 0, + "shapes_from": "unseen" + }, + "orientations": { + "max": 6.28318531, + "min": 0.0, + "num": 40 + }, + "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "output_channels": 1, + "position_space": "renderer", + "positions_x": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "positions_y": { + "max": 0.8, + "min": 0.2, + "num": 32 + }, + "scale_space": "renderer", + "scales": [ + 0.2, + 0.24, + 0.28, + 0.32, + 0.36, + 0.4 + ], + "seed": 42, + "shape_pool": "all", + "train": { + "num_bg_colors": 1, + "num_fg_colors": 1, + "num_shapes": 25 + } + }, + "has_factor_structure": false, + "image_size": 64, + "num_images": 6144000, + "num_values": 2, + "output_channels": 1, + "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt", + "split_config": { + "bg_colors_from": "unseen", + "fg_colors_from": "unseen", + "ood_bg_colors": [], + "ood_fg_colors": [], + "ood_index_range": { + "start": 6144000, + "stop": 6144000 + }, + "ood_shapes": [], + "ood_size": 0, + "seed": 42, + "shapes_from": "unseen", + "train_bg_colors": [ + [ + 0, + 0, + 0 + ] + ], + "train_fg_colors": [ + [ + 1, + 1, + 1 + ] + ], + "train_index_range": { + "start": 0, + "stop": 6144000 + }, + "train_shapes": [ + "polygon_3", + "polygon_4", + "polygon_5", + "star_3_r02", + "star_4_r04", + "star_5_r04", + "star_6_r04", + "ellipse_050", + "heart", + "flower_2_w03", + "flower_3_w03", + "flower_4_w03", + "flower_5_w03", + "flower_6_w03", + "rectangle_03", + "cross_065", + "crescent_025", + "drop_15", + "closed_hollow_polygon_3", + "closed_hollow_polygon_4", + "closed_hollow_polygon_5", + "hollow_polygon_3", + "hollow_polygon_4", + "hollow_polygon_5", + "closed_hollow_ellipse_05" + ], + "train_size": 6144000 + } + }, + "composition": { + "action_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ], + "command_sampler": "lils", + "held_out_ood_size": 122880, + "ignored_factors": [ + "foreground_color", + "background_color" + ], + "in_domain_pool_size": 614400, + "ood_excluded_from_pool": true, + "ood_excluded_from_training_pool": true, + "ood_validation_pool_size": 737280, + "ood_validation_uses_combined_pool": true, + "split_source": "restricted_shape_holdout", + "task_factors": [ + "shape", + "scale", + "orientation", + "position_x", + "position_y" + ] + }, + "dataset_name": "sprites_large_composition", + "loader_filter": { + "seed": 2389822288, + "sprite_restricted_shapes": [ + "flower_6_w03" + ], + "sprite_right_side_threshold": 0.5, + "sprite_shapes": [ + "flower_6_w03", + "flower_5_w03", + "cross_065" + ], + "val_fraction": 0.0 + }, + "splits": { + "held_out_ood_size": 122880, + "in_domain_pool_size": 614400, + "ood_validation_pool_size": 737280, + "train_anchor_size": 614399, + "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.", + "val_anchor_size": 1, + "val_in_domain_anchor_size": 1, + "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.", + "val_ood_anchor_size": 122880, + "val_ood_rule": "Validate composition on fixed held-out OOD anchors; command and target images are sampled from the combined in-domain+OOD pool so all enabled actions remain evaluable.", + "val_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles." + } +} \ No newline at end of file diff --git a/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775646993.node04.1274707.1 b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775646993.node04.1274707.1 new file mode 100644 index 0000000000000000000000000000000000000000..09b9604598e1949bfd9e612dc87e7273ab6f0690 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775646993.node04.1274707.1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1538ffa020f037656bf3cd965c617c5418cbe8f4603fdda38ba0c3ad02173c1a +size 137657 diff --git a/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml new file mode 100644 index 0000000000000000000000000000000000000000..586ed1119a20259cbf432889d19b59f43ec4ab72 --- /dev/null +++ b/CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml @@ -0,0 +1,10 @@ +num_filters: 64 +z_dim: 10 +lr: 0.0001 +composition_op: fixint +n_actions: 5 +num_channels: 1 +num_values: 2 +image_size: 64 +bbox_loss_inside_weight: 1.0 +bbox_loss_outside_weight: 1.0