Sync log files from VAE_logs and CompVAE_logs
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.json +156 -0
- CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.yaml +131 -0
- CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json +142 -0
- CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml +118 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoint_summary.json +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/dataset_loader_config.json +212 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/events.out.tfevents.1775565498.node04.1267668.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/dataset_loader_config.json +212 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/events.out.tfevents.1775565909.node05.1313440.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoint_summary.json +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json +212 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775566849.node04.1274707.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/dataset_loader_config.json +215 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/events.out.tfevents.1775566847.node05.1316752.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.json +149 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.yaml +124 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/dataset_loader_config.json +218 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/events.out.tfevents.1775566862.node06.2124470.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.json +155 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.yaml +130 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/dataset_loader_config.json +221 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/events.out.tfevents.1775566873.node06.2124623.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.json +161 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.yaml +136 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/dataset_loader_config.json +224 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/events.out.tfevents.1775566873.node06.2124788.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.json +167 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.yaml +142 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/dataset_loader_config.json +229 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/events.out.tfevents.1775566879.node07.1629786.0 +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/hparams.yaml +10 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.json +177 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.yaml +152 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/checkpoints/best_in_domain.ckpt +3 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/dataset_loader_config.json +234 -0
- CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/events.out.tfevents.1775566899.node07.1629937.0 +3 -0
CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.json
ADDED
|
@@ -0,0 +1,156 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"available_shapes": [
|
| 3 |
+
"polygon_3",
|
| 4 |
+
"polygon_4",
|
| 5 |
+
"polygon_5",
|
| 6 |
+
"star_3_r02",
|
| 7 |
+
"star_4_r04",
|
| 8 |
+
"star_5_r04",
|
| 9 |
+
"star_6_r04",
|
| 10 |
+
"ellipse_050",
|
| 11 |
+
"heart",
|
| 12 |
+
"flower_2_w03",
|
| 13 |
+
"flower_3_w03",
|
| 14 |
+
"flower_4_w03",
|
| 15 |
+
"flower_5_w03",
|
| 16 |
+
"flower_6_w03",
|
| 17 |
+
"rectangle_03",
|
| 18 |
+
"cross_065",
|
| 19 |
+
"crescent_025",
|
| 20 |
+
"drop_15",
|
| 21 |
+
"closed_hollow_polygon_3",
|
| 22 |
+
"closed_hollow_polygon_4",
|
| 23 |
+
"closed_hollow_polygon_5",
|
| 24 |
+
"hollow_polygon_3",
|
| 25 |
+
"hollow_polygon_4",
|
| 26 |
+
"hollow_polygon_5",
|
| 27 |
+
"closed_hollow_ellipse_05"
|
| 28 |
+
],
|
| 29 |
+
"base_generation_config": {
|
| 30 |
+
"candidate_bg_colors": [
|
| 31 |
+
[
|
| 32 |
+
0,
|
| 33 |
+
0,
|
| 34 |
+
0
|
| 35 |
+
]
|
| 36 |
+
],
|
| 37 |
+
"candidate_fg_colors": [
|
| 38 |
+
[
|
| 39 |
+
255,
|
| 40 |
+
255,
|
| 41 |
+
255
|
| 42 |
+
]
|
| 43 |
+
],
|
| 44 |
+
"image_size": 64,
|
| 45 |
+
"num_values": 4,
|
| 46 |
+
"ood": {
|
| 47 |
+
"bg_colors_from": "unseen",
|
| 48 |
+
"fg_colors_from": "unseen",
|
| 49 |
+
"num_bg_colors": 0,
|
| 50 |
+
"num_fg_colors": 0,
|
| 51 |
+
"num_shapes": 0,
|
| 52 |
+
"shapes_from": "unseen"
|
| 53 |
+
},
|
| 54 |
+
"orientations": {
|
| 55 |
+
"max": 6.28318531,
|
| 56 |
+
"min": 0.0,
|
| 57 |
+
"num": 20
|
| 58 |
+
},
|
| 59 |
+
"output": "data/lils/dsprites_large_6_20_16_16.pt",
|
| 60 |
+
"output_channels": 1,
|
| 61 |
+
"position_space": "renderer",
|
| 62 |
+
"positions_x": {
|
| 63 |
+
"max": 0.8,
|
| 64 |
+
"min": 0.2,
|
| 65 |
+
"num": 16
|
| 66 |
+
},
|
| 67 |
+
"positions_y": {
|
| 68 |
+
"max": 0.8,
|
| 69 |
+
"min": 0.2,
|
| 70 |
+
"num": 16
|
| 71 |
+
},
|
| 72 |
+
"scale_space": "renderer",
|
| 73 |
+
"scales": [
|
| 74 |
+
0.2,
|
| 75 |
+
0.24,
|
| 76 |
+
0.28,
|
| 77 |
+
0.32,
|
| 78 |
+
0.36,
|
| 79 |
+
0.4
|
| 80 |
+
],
|
| 81 |
+
"seed": 42,
|
| 82 |
+
"shape_pool": "all",
|
| 83 |
+
"train": {
|
| 84 |
+
"num_bg_colors": 1,
|
| 85 |
+
"num_fg_colors": 1,
|
| 86 |
+
"num_shapes": 25
|
| 87 |
+
}
|
| 88 |
+
},
|
| 89 |
+
"composition": {
|
| 90 |
+
"action_factors": [
|
| 91 |
+
"foreground_color",
|
| 92 |
+
"background_color",
|
| 93 |
+
"shape",
|
| 94 |
+
"scale",
|
| 95 |
+
"orientation",
|
| 96 |
+
"position_x",
|
| 97 |
+
"position_y"
|
| 98 |
+
],
|
| 99 |
+
"command_sampler": "uniform",
|
| 100 |
+
"composition_op": "interp",
|
| 101 |
+
"ignored_factors": []
|
| 102 |
+
},
|
| 103 |
+
"n_shapes_sweep": [
|
| 104 |
+
3
|
| 105 |
+
],
|
| 106 |
+
"num_reruns": 1,
|
| 107 |
+
"rerun_offset": 0,
|
| 108 |
+
"runs": [
|
| 109 |
+
{
|
| 110 |
+
"held_out_shape": "polygon_3",
|
| 111 |
+
"n_shapes": 3,
|
| 112 |
+
"rerun_index": 0,
|
| 113 |
+
"run_index": 0,
|
| 114 |
+
"run_seed": 2548879074,
|
| 115 |
+
"run_shapes": [
|
| 116 |
+
"polygon_3",
|
| 117 |
+
"polygon_5",
|
| 118 |
+
"star_5_r04"
|
| 119 |
+
],
|
| 120 |
+
"z_dim": 10
|
| 121 |
+
}
|
| 122 |
+
],
|
| 123 |
+
"shape_pool": [
|
| 124 |
+
"polygon_3",
|
| 125 |
+
"polygon_4",
|
| 126 |
+
"polygon_5",
|
| 127 |
+
"star_3_r02",
|
| 128 |
+
"star_4_r04",
|
| 129 |
+
"star_5_r04",
|
| 130 |
+
"star_6_r04",
|
| 131 |
+
"ellipse_050",
|
| 132 |
+
"heart",
|
| 133 |
+
"flower_2_w03",
|
| 134 |
+
"flower_3_w03",
|
| 135 |
+
"flower_4_w03",
|
| 136 |
+
"flower_5_w03",
|
| 137 |
+
"flower_6_w03",
|
| 138 |
+
"rectangle_03",
|
| 139 |
+
"cross_065",
|
| 140 |
+
"crescent_025",
|
| 141 |
+
"drop_15",
|
| 142 |
+
"closed_hollow_polygon_3",
|
| 143 |
+
"closed_hollow_polygon_4",
|
| 144 |
+
"closed_hollow_polygon_5",
|
| 145 |
+
"hollow_polygon_3",
|
| 146 |
+
"hollow_polygon_4",
|
| 147 |
+
"hollow_polygon_5",
|
| 148 |
+
"closed_hollow_ellipse_05"
|
| 149 |
+
],
|
| 150 |
+
"sprite_right_side_threshold": 0.5,
|
| 151 |
+
"sprites_large_cache": "data/lils/dsprites_large_6_20_16_16.pt",
|
| 152 |
+
"val_fraction": 0.0,
|
| 153 |
+
"z_dims": [
|
| 154 |
+
10
|
| 155 |
+
]
|
| 156 |
+
}
|
CompVAE_logs/dsprites_composition_sweep_lils_fixint/composition_sweep_plan.yaml
ADDED
|
@@ -0,0 +1,131 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
sprites_large_cache: data/lils/dsprites_large_6_20_16_16.pt
|
| 2 |
+
available_shapes:
|
| 3 |
+
- polygon_3
|
| 4 |
+
- polygon_4
|
| 5 |
+
- polygon_5
|
| 6 |
+
- star_3_r02
|
| 7 |
+
- star_4_r04
|
| 8 |
+
- star_5_r04
|
| 9 |
+
- star_6_r04
|
| 10 |
+
- ellipse_050
|
| 11 |
+
- heart
|
| 12 |
+
- flower_2_w03
|
| 13 |
+
- flower_3_w03
|
| 14 |
+
- flower_4_w03
|
| 15 |
+
- flower_5_w03
|
| 16 |
+
- flower_6_w03
|
| 17 |
+
- rectangle_03
|
| 18 |
+
- cross_065
|
| 19 |
+
- crescent_025
|
| 20 |
+
- drop_15
|
| 21 |
+
- closed_hollow_polygon_3
|
| 22 |
+
- closed_hollow_polygon_4
|
| 23 |
+
- closed_hollow_polygon_5
|
| 24 |
+
- hollow_polygon_3
|
| 25 |
+
- hollow_polygon_4
|
| 26 |
+
- hollow_polygon_5
|
| 27 |
+
- closed_hollow_ellipse_05
|
| 28 |
+
shape_pool:
|
| 29 |
+
- polygon_3
|
| 30 |
+
- polygon_4
|
| 31 |
+
- polygon_5
|
| 32 |
+
- star_3_r02
|
| 33 |
+
- star_4_r04
|
| 34 |
+
- star_5_r04
|
| 35 |
+
- star_6_r04
|
| 36 |
+
- ellipse_050
|
| 37 |
+
- heart
|
| 38 |
+
- flower_2_w03
|
| 39 |
+
- flower_3_w03
|
| 40 |
+
- flower_4_w03
|
| 41 |
+
- flower_5_w03
|
| 42 |
+
- flower_6_w03
|
| 43 |
+
- rectangle_03
|
| 44 |
+
- cross_065
|
| 45 |
+
- crescent_025
|
| 46 |
+
- drop_15
|
| 47 |
+
- closed_hollow_polygon_3
|
| 48 |
+
- closed_hollow_polygon_4
|
| 49 |
+
- closed_hollow_polygon_5
|
| 50 |
+
- hollow_polygon_3
|
| 51 |
+
- hollow_polygon_4
|
| 52 |
+
- hollow_polygon_5
|
| 53 |
+
- closed_hollow_ellipse_05
|
| 54 |
+
num_reruns: 1
|
| 55 |
+
rerun_offset: 0
|
| 56 |
+
sprite_right_side_threshold: 0.5
|
| 57 |
+
val_fraction: 0.0
|
| 58 |
+
n_shapes_sweep:
|
| 59 |
+
- 3
|
| 60 |
+
z_dims:
|
| 61 |
+
- 10
|
| 62 |
+
base_generation_config:
|
| 63 |
+
seed: 42
|
| 64 |
+
image_size: 64
|
| 65 |
+
num_values: 4
|
| 66 |
+
output_channels: 1
|
| 67 |
+
output: data/lils/dsprites_large_6_20_16_16.pt
|
| 68 |
+
scale_space: renderer
|
| 69 |
+
position_space: renderer
|
| 70 |
+
shape_pool: all
|
| 71 |
+
candidate_fg_colors:
|
| 72 |
+
- - 255
|
| 73 |
+
- 255
|
| 74 |
+
- 255
|
| 75 |
+
candidate_bg_colors:
|
| 76 |
+
- - 0
|
| 77 |
+
- 0
|
| 78 |
+
- 0
|
| 79 |
+
scales:
|
| 80 |
+
- 0.2
|
| 81 |
+
- 0.24
|
| 82 |
+
- 0.28
|
| 83 |
+
- 0.32
|
| 84 |
+
- 0.36
|
| 85 |
+
- 0.4
|
| 86 |
+
orientations:
|
| 87 |
+
min: 0.0
|
| 88 |
+
max: 6.28318531
|
| 89 |
+
num: 20
|
| 90 |
+
positions_x:
|
| 91 |
+
min: 0.2
|
| 92 |
+
max: 0.8
|
| 93 |
+
num: 16
|
| 94 |
+
positions_y:
|
| 95 |
+
min: 0.2
|
| 96 |
+
max: 0.8
|
| 97 |
+
num: 16
|
| 98 |
+
train:
|
| 99 |
+
num_shapes: 25
|
| 100 |
+
num_fg_colors: 1
|
| 101 |
+
num_bg_colors: 1
|
| 102 |
+
ood:
|
| 103 |
+
num_shapes: 0
|
| 104 |
+
shapes_from: unseen
|
| 105 |
+
num_fg_colors: 0
|
| 106 |
+
fg_colors_from: unseen
|
| 107 |
+
num_bg_colors: 0
|
| 108 |
+
bg_colors_from: unseen
|
| 109 |
+
composition:
|
| 110 |
+
composition_op: interp
|
| 111 |
+
command_sampler: uniform
|
| 112 |
+
action_factors:
|
| 113 |
+
- foreground_color
|
| 114 |
+
- background_color
|
| 115 |
+
- shape
|
| 116 |
+
- scale
|
| 117 |
+
- orientation
|
| 118 |
+
- position_x
|
| 119 |
+
- position_y
|
| 120 |
+
ignored_factors: []
|
| 121 |
+
runs:
|
| 122 |
+
- run_index: 0
|
| 123 |
+
rerun_index: 0
|
| 124 |
+
run_seed: 2548879074
|
| 125 |
+
n_shapes: 3
|
| 126 |
+
z_dim: 10
|
| 127 |
+
held_out_shape: polygon_3
|
| 128 |
+
run_shapes:
|
| 129 |
+
- polygon_3
|
| 130 |
+
- polygon_5
|
| 131 |
+
- star_5_r04
|
CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json
ADDED
|
@@ -0,0 +1,142 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 4,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 20
|
| 59 |
+
},
|
| 60 |
+
"output": "data/lils/dsprites_large_6_20_16_16.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 16
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 16
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"path": "data/lils/dsprites_large_6_20_16_16.pt"
|
| 91 |
+
},
|
| 92 |
+
"composition": {
|
| 93 |
+
"action_factors": [
|
| 94 |
+
"foreground_color",
|
| 95 |
+
"background_color",
|
| 96 |
+
"shape",
|
| 97 |
+
"scale",
|
| 98 |
+
"orientation",
|
| 99 |
+
"position_x",
|
| 100 |
+
"position_y"
|
| 101 |
+
],
|
| 102 |
+
"command_sampler": "uniform",
|
| 103 |
+
"composition_op": "interp",
|
| 104 |
+
"ignored_factors": []
|
| 105 |
+
},
|
| 106 |
+
"dataloader": {
|
| 107 |
+
"sprite_restricted_shapes": [
|
| 108 |
+
"polygon_3"
|
| 109 |
+
],
|
| 110 |
+
"sprite_right_side_threshold": 0.5,
|
| 111 |
+
"sprite_shapes": [
|
| 112 |
+
"polygon_3",
|
| 113 |
+
"polygon_5",
|
| 114 |
+
"star_5_r04"
|
| 115 |
+
],
|
| 116 |
+
"val_fraction": 0.0
|
| 117 |
+
},
|
| 118 |
+
"held_out_shape": "polygon_3",
|
| 119 |
+
"n_shapes": 3,
|
| 120 |
+
"rerun_index": 0,
|
| 121 |
+
"run_index": 0,
|
| 122 |
+
"run_name": "run_000_rerun_00_03shapes_zdim_010",
|
| 123 |
+
"run_seed": 2548879074,
|
| 124 |
+
"run_shapes": [
|
| 125 |
+
"polygon_3",
|
| 126 |
+
"polygon_5",
|
| 127 |
+
"star_5_r04"
|
| 128 |
+
],
|
| 129 |
+
"training": {
|
| 130 |
+
"base_seed": 42,
|
| 131 |
+
"batch_size": 128,
|
| 132 |
+
"bbox_loss_inside_weight": 9.0,
|
| 133 |
+
"bbox_loss_outside_weight": 1.0,
|
| 134 |
+
"epochs": 1,
|
| 135 |
+
"lr": 0.0001,
|
| 136 |
+
"num_filters": 64,
|
| 137 |
+
"num_values": 2,
|
| 138 |
+
"num_workers": 4,
|
| 139 |
+
"seed": 2548879074
|
| 140 |
+
},
|
| 141 |
+
"z_dim": 10
|
| 142 |
+
}
|
CompVAE_logs/dsprites_composition_sweep_lils_fixint/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml
ADDED
|
@@ -0,0 +1,118 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_index: 0
|
| 2 |
+
rerun_index: 0
|
| 3 |
+
run_name: run_000_rerun_00_03shapes_zdim_010
|
| 4 |
+
run_seed: 2548879074
|
| 5 |
+
z_dim: 10
|
| 6 |
+
n_shapes: 3
|
| 7 |
+
held_out_shape: polygon_3
|
| 8 |
+
run_shapes:
|
| 9 |
+
- polygon_3
|
| 10 |
+
- polygon_5
|
| 11 |
+
- star_5_r04
|
| 12 |
+
cache:
|
| 13 |
+
path: data/lils/dsprites_large_6_20_16_16.pt
|
| 14 |
+
available_shapes:
|
| 15 |
+
- polygon_3
|
| 16 |
+
- polygon_4
|
| 17 |
+
- polygon_5
|
| 18 |
+
- star_3_r02
|
| 19 |
+
- star_4_r04
|
| 20 |
+
- star_5_r04
|
| 21 |
+
- star_6_r04
|
| 22 |
+
- ellipse_050
|
| 23 |
+
- heart
|
| 24 |
+
- flower_2_w03
|
| 25 |
+
- flower_3_w03
|
| 26 |
+
- flower_4_w03
|
| 27 |
+
- flower_5_w03
|
| 28 |
+
- flower_6_w03
|
| 29 |
+
- rectangle_03
|
| 30 |
+
- cross_065
|
| 31 |
+
- crescent_025
|
| 32 |
+
- drop_15
|
| 33 |
+
- closed_hollow_polygon_3
|
| 34 |
+
- closed_hollow_polygon_4
|
| 35 |
+
- closed_hollow_polygon_5
|
| 36 |
+
- hollow_polygon_3
|
| 37 |
+
- hollow_polygon_4
|
| 38 |
+
- hollow_polygon_5
|
| 39 |
+
- closed_hollow_ellipse_05
|
| 40 |
+
generation_config:
|
| 41 |
+
seed: 42
|
| 42 |
+
image_size: 64
|
| 43 |
+
num_values: 4
|
| 44 |
+
output_channels: 1
|
| 45 |
+
output: data/lils/dsprites_large_6_20_16_16.pt
|
| 46 |
+
scale_space: renderer
|
| 47 |
+
position_space: renderer
|
| 48 |
+
shape_pool: all
|
| 49 |
+
candidate_fg_colors:
|
| 50 |
+
- - 255
|
| 51 |
+
- 255
|
| 52 |
+
- 255
|
| 53 |
+
candidate_bg_colors:
|
| 54 |
+
- - 0
|
| 55 |
+
- 0
|
| 56 |
+
- 0
|
| 57 |
+
scales:
|
| 58 |
+
- 0.2
|
| 59 |
+
- 0.24
|
| 60 |
+
- 0.28
|
| 61 |
+
- 0.32
|
| 62 |
+
- 0.36
|
| 63 |
+
- 0.4
|
| 64 |
+
orientations:
|
| 65 |
+
min: 0.0
|
| 66 |
+
max: 6.28318531
|
| 67 |
+
num: 20
|
| 68 |
+
positions_x:
|
| 69 |
+
min: 0.2
|
| 70 |
+
max: 0.8
|
| 71 |
+
num: 16
|
| 72 |
+
positions_y:
|
| 73 |
+
min: 0.2
|
| 74 |
+
max: 0.8
|
| 75 |
+
num: 16
|
| 76 |
+
train:
|
| 77 |
+
num_shapes: 25
|
| 78 |
+
num_fg_colors: 1
|
| 79 |
+
num_bg_colors: 1
|
| 80 |
+
ood:
|
| 81 |
+
num_shapes: 0
|
| 82 |
+
shapes_from: unseen
|
| 83 |
+
num_fg_colors: 0
|
| 84 |
+
fg_colors_from: unseen
|
| 85 |
+
num_bg_colors: 0
|
| 86 |
+
bg_colors_from: unseen
|
| 87 |
+
dataloader:
|
| 88 |
+
sprite_shapes:
|
| 89 |
+
- polygon_3
|
| 90 |
+
- polygon_5
|
| 91 |
+
- star_5_r04
|
| 92 |
+
sprite_restricted_shapes:
|
| 93 |
+
- polygon_3
|
| 94 |
+
sprite_right_side_threshold: 0.5
|
| 95 |
+
val_fraction: 0.0
|
| 96 |
+
composition:
|
| 97 |
+
composition_op: interp
|
| 98 |
+
command_sampler: uniform
|
| 99 |
+
action_factors:
|
| 100 |
+
- foreground_color
|
| 101 |
+
- background_color
|
| 102 |
+
- shape
|
| 103 |
+
- scale
|
| 104 |
+
- orientation
|
| 105 |
+
- position_x
|
| 106 |
+
- position_y
|
| 107 |
+
ignored_factors: []
|
| 108 |
+
training:
|
| 109 |
+
epochs: 1
|
| 110 |
+
batch_size: 128
|
| 111 |
+
num_workers: 4
|
| 112 |
+
num_values: 2
|
| 113 |
+
num_filters: 64
|
| 114 |
+
lr: 0.0001
|
| 115 |
+
seed: 2548879074
|
| 116 |
+
base_seed: 42
|
| 117 |
+
bbox_loss_inside_weight: 9.0
|
| 118 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoint_summary.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"checkpoints": {
|
| 3 |
+
"best_in_domain": {
|
| 4 |
+
"monitor": "val_in_domain_bpd",
|
| 5 |
+
"path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt",
|
| 6 |
+
"score": 0.04238998889923096
|
| 7 |
+
}
|
| 8 |
+
},
|
| 9 |
+
"default_checkpoint": "best_in_domain"
|
| 10 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0ca8d9925c4c94bf295097f74c737103299dcd8a910a4bd8fa8161107b12a46
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,212 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32_mmap",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 614400,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 737280,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 2548879074,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"polygon_3"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"polygon_3",
|
| 194 |
+
"polygon_5",
|
| 195 |
+
"star_5_r04"
|
| 196 |
+
],
|
| 197 |
+
"val_fraction": 0.0
|
| 198 |
+
},
|
| 199 |
+
"splits": {
|
| 200 |
+
"held_out_ood_size": 122880,
|
| 201 |
+
"in_domain_pool_size": 614400,
|
| 202 |
+
"ood_validation_pool_size": 737280,
|
| 203 |
+
"train_anchor_size": 614399,
|
| 204 |
+
"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.",
|
| 205 |
+
"val_anchor_size": 1,
|
| 206 |
+
"val_in_domain_anchor_size": 1,
|
| 207 |
+
"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.",
|
| 208 |
+
"val_ood_anchor_size": 122880,
|
| 209 |
+
"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.",
|
| 210 |
+
"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."
|
| 211 |
+
}
|
| 212 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/events.out.tfevents.1775565498.node04.1267668.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d18f0a7b81e8594fb37720ca4d22362b4aea6176da23a107c2ccc8f86034804c
|
| 3 |
+
size 62058
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12446/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,212 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 614400,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 737280,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 2548879074,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"polygon_3"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"polygon_3",
|
| 194 |
+
"polygon_5",
|
| 195 |
+
"star_5_r04"
|
| 196 |
+
],
|
| 197 |
+
"val_fraction": 0.0
|
| 198 |
+
},
|
| 199 |
+
"splits": {
|
| 200 |
+
"held_out_ood_size": 122880,
|
| 201 |
+
"in_domain_pool_size": 614400,
|
| 202 |
+
"ood_validation_pool_size": 737280,
|
| 203 |
+
"train_anchor_size": 614399,
|
| 204 |
+
"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.",
|
| 205 |
+
"val_anchor_size": 1,
|
| 206 |
+
"val_in_domain_anchor_size": 1,
|
| 207 |
+
"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.",
|
| 208 |
+
"val_ood_anchor_size": 122880,
|
| 209 |
+
"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.",
|
| 210 |
+
"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."
|
| 211 |
+
}
|
| 212 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/events.out.tfevents.1775565909.node05.1313440.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d4c433d7cbc3f1154774af5c022ac73b58c9e572f53d73ce2e98dd596e50725c
|
| 3 |
+
size 5466
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12448/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoint_summary.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"checkpoints": {
|
| 3 |
+
"best_in_domain": {
|
| 4 |
+
"monitor": "val_in_domain_bpd",
|
| 5 |
+
"path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt",
|
| 6 |
+
"score": 0.006012121215462685
|
| 7 |
+
}
|
| 8 |
+
},
|
| 9 |
+
"default_checkpoint": "best_in_domain"
|
| 10 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:726558b91b4007537058f3b4710b1b77f2a7088098465cf3d0271ef34c909bab
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,212 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 614400,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 737280,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 2548879074,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"polygon_3"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"polygon_3",
|
| 194 |
+
"polygon_5",
|
| 195 |
+
"star_5_r04"
|
| 196 |
+
],
|
| 197 |
+
"val_fraction": 0.0
|
| 198 |
+
},
|
| 199 |
+
"splits": {
|
| 200 |
+
"held_out_ood_size": 122880,
|
| 201 |
+
"in_domain_pool_size": 614400,
|
| 202 |
+
"ood_validation_pool_size": 737280,
|
| 203 |
+
"train_anchor_size": 614399,
|
| 204 |
+
"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.",
|
| 205 |
+
"val_anchor_size": 1,
|
| 206 |
+
"val_in_domain_anchor_size": 1,
|
| 207 |
+
"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.",
|
| 208 |
+
"val_ood_anchor_size": 122880,
|
| 209 |
+
"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.",
|
| 210 |
+
"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."
|
| 211 |
+
}
|
| 212 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/events.out.tfevents.1775566849.node04.1274707.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:56aeb8f35f22f859c36ed0b18a37efd9a4199c5384d33caf1996d2eedcc61c7c
|
| 3 |
+
size 693700
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12451/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:91df25f9a83df4fd5719e84e782c5fc44e1814c0a32a01974f54614e974c97be
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,215 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 1351680,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 1474560,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 3003674970,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"star_3_r02"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"star_3_r02",
|
| 194 |
+
"star_6_r04",
|
| 195 |
+
"heart",
|
| 196 |
+
"cross_065",
|
| 197 |
+
"crescent_025",
|
| 198 |
+
"closed_hollow_polygon_3"
|
| 199 |
+
],
|
| 200 |
+
"val_fraction": 0.0
|
| 201 |
+
},
|
| 202 |
+
"splits": {
|
| 203 |
+
"held_out_ood_size": 122880,
|
| 204 |
+
"in_domain_pool_size": 1351680,
|
| 205 |
+
"ood_validation_pool_size": 1474560,
|
| 206 |
+
"train_anchor_size": 1351679,
|
| 207 |
+
"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.",
|
| 208 |
+
"val_anchor_size": 1,
|
| 209 |
+
"val_in_domain_anchor_size": 1,
|
| 210 |
+
"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.",
|
| 211 |
+
"val_ood_anchor_size": 122880,
|
| 212 |
+
"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.",
|
| 213 |
+
"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."
|
| 214 |
+
}
|
| 215 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/events.out.tfevents.1775566847.node05.1316752.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:feed6a4c297cb46b2d591ed70fd109e973438bf21d14905ea073dff3d08a751e
|
| 3 |
+
size 411229
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12452/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.json
ADDED
|
@@ -0,0 +1,149 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
|
| 91 |
+
},
|
| 92 |
+
"composition": {
|
| 93 |
+
"action_factors": [
|
| 94 |
+
"shape",
|
| 95 |
+
"scale",
|
| 96 |
+
"orientation",
|
| 97 |
+
"position_x",
|
| 98 |
+
"position_y"
|
| 99 |
+
],
|
| 100 |
+
"command_sampler": "lils",
|
| 101 |
+
"composition_op": "fixint",
|
| 102 |
+
"ignored_factors": [
|
| 103 |
+
"foreground_color",
|
| 104 |
+
"background_color"
|
| 105 |
+
]
|
| 106 |
+
},
|
| 107 |
+
"dataloader": {
|
| 108 |
+
"sprite_restricted_shapes": [
|
| 109 |
+
"star_3_r02"
|
| 110 |
+
],
|
| 111 |
+
"sprite_right_side_threshold": 0.5,
|
| 112 |
+
"sprite_shapes": [
|
| 113 |
+
"star_3_r02",
|
| 114 |
+
"star_6_r04",
|
| 115 |
+
"heart",
|
| 116 |
+
"cross_065",
|
| 117 |
+
"crescent_025",
|
| 118 |
+
"closed_hollow_polygon_3"
|
| 119 |
+
],
|
| 120 |
+
"val_fraction": 0.0
|
| 121 |
+
},
|
| 122 |
+
"held_out_shape": "star_3_r02",
|
| 123 |
+
"n_shapes": 6,
|
| 124 |
+
"rerun_index": 0,
|
| 125 |
+
"run_index": 0,
|
| 126 |
+
"run_name": "run_000_rerun_00_06shapes_zdim_010",
|
| 127 |
+
"run_seed": 3003674970,
|
| 128 |
+
"run_shapes": [
|
| 129 |
+
"star_3_r02",
|
| 130 |
+
"star_6_r04",
|
| 131 |
+
"heart",
|
| 132 |
+
"cross_065",
|
| 133 |
+
"crescent_025",
|
| 134 |
+
"closed_hollow_polygon_3"
|
| 135 |
+
],
|
| 136 |
+
"training": {
|
| 137 |
+
"base_seed": 42,
|
| 138 |
+
"batch_size": 128,
|
| 139 |
+
"bbox_loss_inside_weight": 1.0,
|
| 140 |
+
"bbox_loss_outside_weight": 1.0,
|
| 141 |
+
"epochs": 100,
|
| 142 |
+
"lr": 0.0001,
|
| 143 |
+
"num_filters": 64,
|
| 144 |
+
"num_values": 2,
|
| 145 |
+
"num_workers": 8,
|
| 146 |
+
"seed": 3003674970
|
| 147 |
+
},
|
| 148 |
+
"z_dim": 10
|
| 149 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_06shapes_zdim_010/sweep_run_config.yaml
ADDED
|
@@ -0,0 +1,124 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_index: 0
|
| 2 |
+
rerun_index: 0
|
| 3 |
+
run_name: run_000_rerun_00_06shapes_zdim_010
|
| 4 |
+
run_seed: 3003674970
|
| 5 |
+
z_dim: 10
|
| 6 |
+
n_shapes: 6
|
| 7 |
+
held_out_shape: star_3_r02
|
| 8 |
+
run_shapes:
|
| 9 |
+
- star_3_r02
|
| 10 |
+
- star_6_r04
|
| 11 |
+
- heart
|
| 12 |
+
- cross_065
|
| 13 |
+
- crescent_025
|
| 14 |
+
- closed_hollow_polygon_3
|
| 15 |
+
cache:
|
| 16 |
+
path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 17 |
+
available_shapes:
|
| 18 |
+
- polygon_3
|
| 19 |
+
- polygon_4
|
| 20 |
+
- polygon_5
|
| 21 |
+
- star_3_r02
|
| 22 |
+
- star_4_r04
|
| 23 |
+
- star_5_r04
|
| 24 |
+
- star_6_r04
|
| 25 |
+
- ellipse_050
|
| 26 |
+
- heart
|
| 27 |
+
- flower_2_w03
|
| 28 |
+
- flower_3_w03
|
| 29 |
+
- flower_4_w03
|
| 30 |
+
- flower_5_w03
|
| 31 |
+
- flower_6_w03
|
| 32 |
+
- rectangle_03
|
| 33 |
+
- cross_065
|
| 34 |
+
- crescent_025
|
| 35 |
+
- drop_15
|
| 36 |
+
- closed_hollow_polygon_3
|
| 37 |
+
- closed_hollow_polygon_4
|
| 38 |
+
- closed_hollow_polygon_5
|
| 39 |
+
- hollow_polygon_3
|
| 40 |
+
- hollow_polygon_4
|
| 41 |
+
- hollow_polygon_5
|
| 42 |
+
- closed_hollow_ellipse_05
|
| 43 |
+
generation_config:
|
| 44 |
+
seed: 42
|
| 45 |
+
image_size: 64
|
| 46 |
+
num_values: 2
|
| 47 |
+
output_channels: 1
|
| 48 |
+
output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 49 |
+
scale_space: renderer
|
| 50 |
+
position_space: renderer
|
| 51 |
+
shape_pool: all
|
| 52 |
+
candidate_fg_colors:
|
| 53 |
+
- - 255
|
| 54 |
+
- 255
|
| 55 |
+
- 255
|
| 56 |
+
candidate_bg_colors:
|
| 57 |
+
- - 0
|
| 58 |
+
- 0
|
| 59 |
+
- 0
|
| 60 |
+
scales:
|
| 61 |
+
- 0.2
|
| 62 |
+
- 0.24
|
| 63 |
+
- 0.28
|
| 64 |
+
- 0.32
|
| 65 |
+
- 0.36
|
| 66 |
+
- 0.4
|
| 67 |
+
orientations:
|
| 68 |
+
min: 0.0
|
| 69 |
+
max: 6.28318531
|
| 70 |
+
num: 40
|
| 71 |
+
positions_x:
|
| 72 |
+
min: 0.2
|
| 73 |
+
max: 0.8
|
| 74 |
+
num: 32
|
| 75 |
+
positions_y:
|
| 76 |
+
min: 0.2
|
| 77 |
+
max: 0.8
|
| 78 |
+
num: 32
|
| 79 |
+
train:
|
| 80 |
+
num_shapes: 25
|
| 81 |
+
num_fg_colors: 1
|
| 82 |
+
num_bg_colors: 1
|
| 83 |
+
ood:
|
| 84 |
+
num_shapes: 0
|
| 85 |
+
shapes_from: unseen
|
| 86 |
+
num_fg_colors: 0
|
| 87 |
+
fg_colors_from: unseen
|
| 88 |
+
num_bg_colors: 0
|
| 89 |
+
bg_colors_from: unseen
|
| 90 |
+
dataloader:
|
| 91 |
+
sprite_shapes:
|
| 92 |
+
- star_3_r02
|
| 93 |
+
- star_6_r04
|
| 94 |
+
- heart
|
| 95 |
+
- cross_065
|
| 96 |
+
- crescent_025
|
| 97 |
+
- closed_hollow_polygon_3
|
| 98 |
+
sprite_restricted_shapes:
|
| 99 |
+
- star_3_r02
|
| 100 |
+
sprite_right_side_threshold: 0.5
|
| 101 |
+
val_fraction: 0.0
|
| 102 |
+
composition:
|
| 103 |
+
composition_op: fixint
|
| 104 |
+
command_sampler: lils
|
| 105 |
+
action_factors:
|
| 106 |
+
- shape
|
| 107 |
+
- scale
|
| 108 |
+
- orientation
|
| 109 |
+
- position_x
|
| 110 |
+
- position_y
|
| 111 |
+
ignored_factors:
|
| 112 |
+
- foreground_color
|
| 113 |
+
- background_color
|
| 114 |
+
training:
|
| 115 |
+
epochs: 100
|
| 116 |
+
batch_size: 128
|
| 117 |
+
num_workers: 8
|
| 118 |
+
num_values: 2
|
| 119 |
+
num_filters: 64
|
| 120 |
+
lr: 0.0001
|
| 121 |
+
seed: 3003674970
|
| 122 |
+
base_seed: 42
|
| 123 |
+
bbox_loss_inside_weight: 1.0
|
| 124 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:845924c16efa05d8df19bb498d1e83ed9a0f015d889d277a8642d54e34088b78
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 2088960,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 2211840,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 4083989091,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"polygon_4"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"polygon_4",
|
| 194 |
+
"polygon_5",
|
| 195 |
+
"star_5_r04",
|
| 196 |
+
"flower_3_w03",
|
| 197 |
+
"flower_4_w03",
|
| 198 |
+
"flower_5_w03",
|
| 199 |
+
"drop_15",
|
| 200 |
+
"hollow_polygon_3",
|
| 201 |
+
"closed_hollow_ellipse_05"
|
| 202 |
+
],
|
| 203 |
+
"val_fraction": 0.0
|
| 204 |
+
},
|
| 205 |
+
"splits": {
|
| 206 |
+
"held_out_ood_size": 122880,
|
| 207 |
+
"in_domain_pool_size": 2088960,
|
| 208 |
+
"ood_validation_pool_size": 2211840,
|
| 209 |
+
"train_anchor_size": 2088959,
|
| 210 |
+
"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.",
|
| 211 |
+
"val_anchor_size": 1,
|
| 212 |
+
"val_in_domain_anchor_size": 1,
|
| 213 |
+
"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.",
|
| 214 |
+
"val_ood_anchor_size": 122880,
|
| 215 |
+
"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.",
|
| 216 |
+
"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."
|
| 217 |
+
}
|
| 218 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/events.out.tfevents.1775566862.node06.2124470.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7a69014b68191fcce0009f3c8fb3dbd6136e3f43a4ca7b4a251304b04faab2f0
|
| 3 |
+
size 292269
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12453/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.json
ADDED
|
@@ -0,0 +1,155 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
|
| 91 |
+
},
|
| 92 |
+
"composition": {
|
| 93 |
+
"action_factors": [
|
| 94 |
+
"shape",
|
| 95 |
+
"scale",
|
| 96 |
+
"orientation",
|
| 97 |
+
"position_x",
|
| 98 |
+
"position_y"
|
| 99 |
+
],
|
| 100 |
+
"command_sampler": "lils",
|
| 101 |
+
"composition_op": "fixint",
|
| 102 |
+
"ignored_factors": [
|
| 103 |
+
"foreground_color",
|
| 104 |
+
"background_color"
|
| 105 |
+
]
|
| 106 |
+
},
|
| 107 |
+
"dataloader": {
|
| 108 |
+
"sprite_restricted_shapes": [
|
| 109 |
+
"polygon_4"
|
| 110 |
+
],
|
| 111 |
+
"sprite_right_side_threshold": 0.5,
|
| 112 |
+
"sprite_shapes": [
|
| 113 |
+
"polygon_4",
|
| 114 |
+
"polygon_5",
|
| 115 |
+
"star_5_r04",
|
| 116 |
+
"flower_3_w03",
|
| 117 |
+
"flower_4_w03",
|
| 118 |
+
"flower_5_w03",
|
| 119 |
+
"drop_15",
|
| 120 |
+
"hollow_polygon_3",
|
| 121 |
+
"closed_hollow_ellipse_05"
|
| 122 |
+
],
|
| 123 |
+
"val_fraction": 0.0
|
| 124 |
+
},
|
| 125 |
+
"held_out_shape": "polygon_4",
|
| 126 |
+
"n_shapes": 9,
|
| 127 |
+
"rerun_index": 0,
|
| 128 |
+
"run_index": 0,
|
| 129 |
+
"run_name": "run_000_rerun_00_09shapes_zdim_010",
|
| 130 |
+
"run_seed": 4083989091,
|
| 131 |
+
"run_shapes": [
|
| 132 |
+
"polygon_4",
|
| 133 |
+
"polygon_5",
|
| 134 |
+
"star_5_r04",
|
| 135 |
+
"flower_3_w03",
|
| 136 |
+
"flower_4_w03",
|
| 137 |
+
"flower_5_w03",
|
| 138 |
+
"drop_15",
|
| 139 |
+
"hollow_polygon_3",
|
| 140 |
+
"closed_hollow_ellipse_05"
|
| 141 |
+
],
|
| 142 |
+
"training": {
|
| 143 |
+
"base_seed": 42,
|
| 144 |
+
"batch_size": 128,
|
| 145 |
+
"bbox_loss_inside_weight": 1.0,
|
| 146 |
+
"bbox_loss_outside_weight": 1.0,
|
| 147 |
+
"epochs": 100,
|
| 148 |
+
"lr": 0.0001,
|
| 149 |
+
"num_filters": 64,
|
| 150 |
+
"num_values": 2,
|
| 151 |
+
"num_workers": 8,
|
| 152 |
+
"seed": 4083989091
|
| 153 |
+
},
|
| 154 |
+
"z_dim": 10
|
| 155 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_09shapes_zdim_010/sweep_run_config.yaml
ADDED
|
@@ -0,0 +1,130 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_index: 0
|
| 2 |
+
rerun_index: 0
|
| 3 |
+
run_name: run_000_rerun_00_09shapes_zdim_010
|
| 4 |
+
run_seed: 4083989091
|
| 5 |
+
z_dim: 10
|
| 6 |
+
n_shapes: 9
|
| 7 |
+
held_out_shape: polygon_4
|
| 8 |
+
run_shapes:
|
| 9 |
+
- polygon_4
|
| 10 |
+
- polygon_5
|
| 11 |
+
- star_5_r04
|
| 12 |
+
- flower_3_w03
|
| 13 |
+
- flower_4_w03
|
| 14 |
+
- flower_5_w03
|
| 15 |
+
- drop_15
|
| 16 |
+
- hollow_polygon_3
|
| 17 |
+
- closed_hollow_ellipse_05
|
| 18 |
+
cache:
|
| 19 |
+
path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 20 |
+
available_shapes:
|
| 21 |
+
- polygon_3
|
| 22 |
+
- polygon_4
|
| 23 |
+
- polygon_5
|
| 24 |
+
- star_3_r02
|
| 25 |
+
- star_4_r04
|
| 26 |
+
- star_5_r04
|
| 27 |
+
- star_6_r04
|
| 28 |
+
- ellipse_050
|
| 29 |
+
- heart
|
| 30 |
+
- flower_2_w03
|
| 31 |
+
- flower_3_w03
|
| 32 |
+
- flower_4_w03
|
| 33 |
+
- flower_5_w03
|
| 34 |
+
- flower_6_w03
|
| 35 |
+
- rectangle_03
|
| 36 |
+
- cross_065
|
| 37 |
+
- crescent_025
|
| 38 |
+
- drop_15
|
| 39 |
+
- closed_hollow_polygon_3
|
| 40 |
+
- closed_hollow_polygon_4
|
| 41 |
+
- closed_hollow_polygon_5
|
| 42 |
+
- hollow_polygon_3
|
| 43 |
+
- hollow_polygon_4
|
| 44 |
+
- hollow_polygon_5
|
| 45 |
+
- closed_hollow_ellipse_05
|
| 46 |
+
generation_config:
|
| 47 |
+
seed: 42
|
| 48 |
+
image_size: 64
|
| 49 |
+
num_values: 2
|
| 50 |
+
output_channels: 1
|
| 51 |
+
output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 52 |
+
scale_space: renderer
|
| 53 |
+
position_space: renderer
|
| 54 |
+
shape_pool: all
|
| 55 |
+
candidate_fg_colors:
|
| 56 |
+
- - 255
|
| 57 |
+
- 255
|
| 58 |
+
- 255
|
| 59 |
+
candidate_bg_colors:
|
| 60 |
+
- - 0
|
| 61 |
+
- 0
|
| 62 |
+
- 0
|
| 63 |
+
scales:
|
| 64 |
+
- 0.2
|
| 65 |
+
- 0.24
|
| 66 |
+
- 0.28
|
| 67 |
+
- 0.32
|
| 68 |
+
- 0.36
|
| 69 |
+
- 0.4
|
| 70 |
+
orientations:
|
| 71 |
+
min: 0.0
|
| 72 |
+
max: 6.28318531
|
| 73 |
+
num: 40
|
| 74 |
+
positions_x:
|
| 75 |
+
min: 0.2
|
| 76 |
+
max: 0.8
|
| 77 |
+
num: 32
|
| 78 |
+
positions_y:
|
| 79 |
+
min: 0.2
|
| 80 |
+
max: 0.8
|
| 81 |
+
num: 32
|
| 82 |
+
train:
|
| 83 |
+
num_shapes: 25
|
| 84 |
+
num_fg_colors: 1
|
| 85 |
+
num_bg_colors: 1
|
| 86 |
+
ood:
|
| 87 |
+
num_shapes: 0
|
| 88 |
+
shapes_from: unseen
|
| 89 |
+
num_fg_colors: 0
|
| 90 |
+
fg_colors_from: unseen
|
| 91 |
+
num_bg_colors: 0
|
| 92 |
+
bg_colors_from: unseen
|
| 93 |
+
dataloader:
|
| 94 |
+
sprite_shapes:
|
| 95 |
+
- polygon_4
|
| 96 |
+
- polygon_5
|
| 97 |
+
- star_5_r04
|
| 98 |
+
- flower_3_w03
|
| 99 |
+
- flower_4_w03
|
| 100 |
+
- flower_5_w03
|
| 101 |
+
- drop_15
|
| 102 |
+
- hollow_polygon_3
|
| 103 |
+
- closed_hollow_ellipse_05
|
| 104 |
+
sprite_restricted_shapes:
|
| 105 |
+
- polygon_4
|
| 106 |
+
sprite_right_side_threshold: 0.5
|
| 107 |
+
val_fraction: 0.0
|
| 108 |
+
composition:
|
| 109 |
+
composition_op: fixint
|
| 110 |
+
command_sampler: lils
|
| 111 |
+
action_factors:
|
| 112 |
+
- shape
|
| 113 |
+
- scale
|
| 114 |
+
- orientation
|
| 115 |
+
- position_x
|
| 116 |
+
- position_y
|
| 117 |
+
ignored_factors:
|
| 118 |
+
- foreground_color
|
| 119 |
+
- background_color
|
| 120 |
+
training:
|
| 121 |
+
epochs: 100
|
| 122 |
+
batch_size: 128
|
| 123 |
+
num_workers: 8
|
| 124 |
+
num_values: 2
|
| 125 |
+
num_filters: 64
|
| 126 |
+
lr: 0.0001
|
| 127 |
+
seed: 4083989091
|
| 128 |
+
base_seed: 42
|
| 129 |
+
bbox_loss_inside_weight: 1.0
|
| 130 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3e24dffeb427c7e9eb7da3993697f22c268420e1c2594fd8d69b2f28b088c08d
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,221 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 2826240,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 2949120,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 941690707,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"closed_hollow_polygon_5"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"closed_hollow_polygon_5",
|
| 194 |
+
"polygon_3",
|
| 195 |
+
"polygon_4",
|
| 196 |
+
"ellipse_050",
|
| 197 |
+
"heart",
|
| 198 |
+
"flower_2_w03",
|
| 199 |
+
"flower_3_w03",
|
| 200 |
+
"flower_4_w03",
|
| 201 |
+
"flower_6_w03",
|
| 202 |
+
"cross_065",
|
| 203 |
+
"drop_15",
|
| 204 |
+
"closed_hollow_polygon_3"
|
| 205 |
+
],
|
| 206 |
+
"val_fraction": 0.0
|
| 207 |
+
},
|
| 208 |
+
"splits": {
|
| 209 |
+
"held_out_ood_size": 122880,
|
| 210 |
+
"in_domain_pool_size": 2826240,
|
| 211 |
+
"ood_validation_pool_size": 2949120,
|
| 212 |
+
"train_anchor_size": 2826239,
|
| 213 |
+
"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.",
|
| 214 |
+
"val_anchor_size": 1,
|
| 215 |
+
"val_in_domain_anchor_size": 1,
|
| 216 |
+
"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.",
|
| 217 |
+
"val_ood_anchor_size": 122880,
|
| 218 |
+
"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.",
|
| 219 |
+
"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."
|
| 220 |
+
}
|
| 221 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/events.out.tfevents.1775566873.node06.2124623.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:676217cc38ca47b224add46fcf6b0c5e258bce43d3cd114413e0a9561de9f5bc
|
| 3 |
+
size 217797
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12454/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.json
ADDED
|
@@ -0,0 +1,161 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
|
| 91 |
+
},
|
| 92 |
+
"composition": {
|
| 93 |
+
"action_factors": [
|
| 94 |
+
"shape",
|
| 95 |
+
"scale",
|
| 96 |
+
"orientation",
|
| 97 |
+
"position_x",
|
| 98 |
+
"position_y"
|
| 99 |
+
],
|
| 100 |
+
"command_sampler": "lils",
|
| 101 |
+
"composition_op": "fixint",
|
| 102 |
+
"ignored_factors": [
|
| 103 |
+
"foreground_color",
|
| 104 |
+
"background_color"
|
| 105 |
+
]
|
| 106 |
+
},
|
| 107 |
+
"dataloader": {
|
| 108 |
+
"sprite_restricted_shapes": [
|
| 109 |
+
"closed_hollow_polygon_5"
|
| 110 |
+
],
|
| 111 |
+
"sprite_right_side_threshold": 0.5,
|
| 112 |
+
"sprite_shapes": [
|
| 113 |
+
"closed_hollow_polygon_5",
|
| 114 |
+
"polygon_3",
|
| 115 |
+
"polygon_4",
|
| 116 |
+
"ellipse_050",
|
| 117 |
+
"heart",
|
| 118 |
+
"flower_2_w03",
|
| 119 |
+
"flower_3_w03",
|
| 120 |
+
"flower_4_w03",
|
| 121 |
+
"flower_6_w03",
|
| 122 |
+
"cross_065",
|
| 123 |
+
"drop_15",
|
| 124 |
+
"closed_hollow_polygon_3"
|
| 125 |
+
],
|
| 126 |
+
"val_fraction": 0.0
|
| 127 |
+
},
|
| 128 |
+
"held_out_shape": "closed_hollow_polygon_5",
|
| 129 |
+
"n_shapes": 12,
|
| 130 |
+
"rerun_index": 0,
|
| 131 |
+
"run_index": 0,
|
| 132 |
+
"run_name": "run_000_rerun_00_12shapes_zdim_010",
|
| 133 |
+
"run_seed": 941690707,
|
| 134 |
+
"run_shapes": [
|
| 135 |
+
"closed_hollow_polygon_5",
|
| 136 |
+
"polygon_3",
|
| 137 |
+
"polygon_4",
|
| 138 |
+
"ellipse_050",
|
| 139 |
+
"heart",
|
| 140 |
+
"flower_2_w03",
|
| 141 |
+
"flower_3_w03",
|
| 142 |
+
"flower_4_w03",
|
| 143 |
+
"flower_6_w03",
|
| 144 |
+
"cross_065",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3"
|
| 147 |
+
],
|
| 148 |
+
"training": {
|
| 149 |
+
"base_seed": 42,
|
| 150 |
+
"batch_size": 128,
|
| 151 |
+
"bbox_loss_inside_weight": 1.0,
|
| 152 |
+
"bbox_loss_outside_weight": 1.0,
|
| 153 |
+
"epochs": 100,
|
| 154 |
+
"lr": 0.0001,
|
| 155 |
+
"num_filters": 64,
|
| 156 |
+
"num_values": 2,
|
| 157 |
+
"num_workers": 8,
|
| 158 |
+
"seed": 941690707
|
| 159 |
+
},
|
| 160 |
+
"z_dim": 10
|
| 161 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_12shapes_zdim_010/sweep_run_config.yaml
ADDED
|
@@ -0,0 +1,136 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_index: 0
|
| 2 |
+
rerun_index: 0
|
| 3 |
+
run_name: run_000_rerun_00_12shapes_zdim_010
|
| 4 |
+
run_seed: 941690707
|
| 5 |
+
z_dim: 10
|
| 6 |
+
n_shapes: 12
|
| 7 |
+
held_out_shape: closed_hollow_polygon_5
|
| 8 |
+
run_shapes:
|
| 9 |
+
- closed_hollow_polygon_5
|
| 10 |
+
- polygon_3
|
| 11 |
+
- polygon_4
|
| 12 |
+
- ellipse_050
|
| 13 |
+
- heart
|
| 14 |
+
- flower_2_w03
|
| 15 |
+
- flower_3_w03
|
| 16 |
+
- flower_4_w03
|
| 17 |
+
- flower_6_w03
|
| 18 |
+
- cross_065
|
| 19 |
+
- drop_15
|
| 20 |
+
- closed_hollow_polygon_3
|
| 21 |
+
cache:
|
| 22 |
+
path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 23 |
+
available_shapes:
|
| 24 |
+
- polygon_3
|
| 25 |
+
- polygon_4
|
| 26 |
+
- polygon_5
|
| 27 |
+
- star_3_r02
|
| 28 |
+
- star_4_r04
|
| 29 |
+
- star_5_r04
|
| 30 |
+
- star_6_r04
|
| 31 |
+
- ellipse_050
|
| 32 |
+
- heart
|
| 33 |
+
- flower_2_w03
|
| 34 |
+
- flower_3_w03
|
| 35 |
+
- flower_4_w03
|
| 36 |
+
- flower_5_w03
|
| 37 |
+
- flower_6_w03
|
| 38 |
+
- rectangle_03
|
| 39 |
+
- cross_065
|
| 40 |
+
- crescent_025
|
| 41 |
+
- drop_15
|
| 42 |
+
- closed_hollow_polygon_3
|
| 43 |
+
- closed_hollow_polygon_4
|
| 44 |
+
- closed_hollow_polygon_5
|
| 45 |
+
- hollow_polygon_3
|
| 46 |
+
- hollow_polygon_4
|
| 47 |
+
- hollow_polygon_5
|
| 48 |
+
- closed_hollow_ellipse_05
|
| 49 |
+
generation_config:
|
| 50 |
+
seed: 42
|
| 51 |
+
image_size: 64
|
| 52 |
+
num_values: 2
|
| 53 |
+
output_channels: 1
|
| 54 |
+
output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 55 |
+
scale_space: renderer
|
| 56 |
+
position_space: renderer
|
| 57 |
+
shape_pool: all
|
| 58 |
+
candidate_fg_colors:
|
| 59 |
+
- - 255
|
| 60 |
+
- 255
|
| 61 |
+
- 255
|
| 62 |
+
candidate_bg_colors:
|
| 63 |
+
- - 0
|
| 64 |
+
- 0
|
| 65 |
+
- 0
|
| 66 |
+
scales:
|
| 67 |
+
- 0.2
|
| 68 |
+
- 0.24
|
| 69 |
+
- 0.28
|
| 70 |
+
- 0.32
|
| 71 |
+
- 0.36
|
| 72 |
+
- 0.4
|
| 73 |
+
orientations:
|
| 74 |
+
min: 0.0
|
| 75 |
+
max: 6.28318531
|
| 76 |
+
num: 40
|
| 77 |
+
positions_x:
|
| 78 |
+
min: 0.2
|
| 79 |
+
max: 0.8
|
| 80 |
+
num: 32
|
| 81 |
+
positions_y:
|
| 82 |
+
min: 0.2
|
| 83 |
+
max: 0.8
|
| 84 |
+
num: 32
|
| 85 |
+
train:
|
| 86 |
+
num_shapes: 25
|
| 87 |
+
num_fg_colors: 1
|
| 88 |
+
num_bg_colors: 1
|
| 89 |
+
ood:
|
| 90 |
+
num_shapes: 0
|
| 91 |
+
shapes_from: unseen
|
| 92 |
+
num_fg_colors: 0
|
| 93 |
+
fg_colors_from: unseen
|
| 94 |
+
num_bg_colors: 0
|
| 95 |
+
bg_colors_from: unseen
|
| 96 |
+
dataloader:
|
| 97 |
+
sprite_shapes:
|
| 98 |
+
- closed_hollow_polygon_5
|
| 99 |
+
- polygon_3
|
| 100 |
+
- polygon_4
|
| 101 |
+
- ellipse_050
|
| 102 |
+
- heart
|
| 103 |
+
- flower_2_w03
|
| 104 |
+
- flower_3_w03
|
| 105 |
+
- flower_4_w03
|
| 106 |
+
- flower_6_w03
|
| 107 |
+
- cross_065
|
| 108 |
+
- drop_15
|
| 109 |
+
- closed_hollow_polygon_3
|
| 110 |
+
sprite_restricted_shapes:
|
| 111 |
+
- closed_hollow_polygon_5
|
| 112 |
+
sprite_right_side_threshold: 0.5
|
| 113 |
+
val_fraction: 0.0
|
| 114 |
+
composition:
|
| 115 |
+
composition_op: fixint
|
| 116 |
+
command_sampler: lils
|
| 117 |
+
action_factors:
|
| 118 |
+
- shape
|
| 119 |
+
- scale
|
| 120 |
+
- orientation
|
| 121 |
+
- position_x
|
| 122 |
+
- position_y
|
| 123 |
+
ignored_factors:
|
| 124 |
+
- foreground_color
|
| 125 |
+
- background_color
|
| 126 |
+
training:
|
| 127 |
+
epochs: 100
|
| 128 |
+
batch_size: 128
|
| 129 |
+
num_workers: 8
|
| 130 |
+
num_values: 2
|
| 131 |
+
num_filters: 64
|
| 132 |
+
lr: 0.0001
|
| 133 |
+
seed: 941690707
|
| 134 |
+
base_seed: 42
|
| 135 |
+
bbox_loss_inside_weight: 1.0
|
| 136 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c64772ef2b3840d306ee4d58794be2d39196f3ae9e20cf50a887739468bd542c
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,224 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 3563520,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 3686400,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 779436189,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"rectangle_03"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"rectangle_03",
|
| 194 |
+
"polygon_3",
|
| 195 |
+
"polygon_4",
|
| 196 |
+
"star_6_r04",
|
| 197 |
+
"ellipse_050",
|
| 198 |
+
"heart",
|
| 199 |
+
"flower_2_w03",
|
| 200 |
+
"flower_5_w03",
|
| 201 |
+
"flower_6_w03",
|
| 202 |
+
"drop_15",
|
| 203 |
+
"closed_hollow_polygon_3",
|
| 204 |
+
"closed_hollow_polygon_4",
|
| 205 |
+
"hollow_polygon_4",
|
| 206 |
+
"hollow_polygon_5",
|
| 207 |
+
"closed_hollow_ellipse_05"
|
| 208 |
+
],
|
| 209 |
+
"val_fraction": 0.0
|
| 210 |
+
},
|
| 211 |
+
"splits": {
|
| 212 |
+
"held_out_ood_size": 122880,
|
| 213 |
+
"in_domain_pool_size": 3563520,
|
| 214 |
+
"ood_validation_pool_size": 3686400,
|
| 215 |
+
"train_anchor_size": 3563519,
|
| 216 |
+
"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.",
|
| 217 |
+
"val_anchor_size": 1,
|
| 218 |
+
"val_in_domain_anchor_size": 1,
|
| 219 |
+
"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.",
|
| 220 |
+
"val_ood_anchor_size": 122880,
|
| 221 |
+
"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.",
|
| 222 |
+
"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."
|
| 223 |
+
}
|
| 224 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/events.out.tfevents.1775566873.node06.2124788.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ecadb835c91765004915392495fd6736b6af7ff6deeb91a597109ae80ece8472
|
| 3 |
+
size 202001
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12455/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.json
ADDED
|
@@ -0,0 +1,167 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
|
| 91 |
+
},
|
| 92 |
+
"composition": {
|
| 93 |
+
"action_factors": [
|
| 94 |
+
"shape",
|
| 95 |
+
"scale",
|
| 96 |
+
"orientation",
|
| 97 |
+
"position_x",
|
| 98 |
+
"position_y"
|
| 99 |
+
],
|
| 100 |
+
"command_sampler": "lils",
|
| 101 |
+
"composition_op": "fixint",
|
| 102 |
+
"ignored_factors": [
|
| 103 |
+
"foreground_color",
|
| 104 |
+
"background_color"
|
| 105 |
+
]
|
| 106 |
+
},
|
| 107 |
+
"dataloader": {
|
| 108 |
+
"sprite_restricted_shapes": [
|
| 109 |
+
"rectangle_03"
|
| 110 |
+
],
|
| 111 |
+
"sprite_right_side_threshold": 0.5,
|
| 112 |
+
"sprite_shapes": [
|
| 113 |
+
"rectangle_03",
|
| 114 |
+
"polygon_3",
|
| 115 |
+
"polygon_4",
|
| 116 |
+
"star_6_r04",
|
| 117 |
+
"ellipse_050",
|
| 118 |
+
"heart",
|
| 119 |
+
"flower_2_w03",
|
| 120 |
+
"flower_5_w03",
|
| 121 |
+
"flower_6_w03",
|
| 122 |
+
"drop_15",
|
| 123 |
+
"closed_hollow_polygon_3",
|
| 124 |
+
"closed_hollow_polygon_4",
|
| 125 |
+
"hollow_polygon_4",
|
| 126 |
+
"hollow_polygon_5",
|
| 127 |
+
"closed_hollow_ellipse_05"
|
| 128 |
+
],
|
| 129 |
+
"val_fraction": 0.0
|
| 130 |
+
},
|
| 131 |
+
"held_out_shape": "rectangle_03",
|
| 132 |
+
"n_shapes": 15,
|
| 133 |
+
"rerun_index": 0,
|
| 134 |
+
"run_index": 0,
|
| 135 |
+
"run_name": "run_000_rerun_00_15shapes_zdim_010",
|
| 136 |
+
"run_seed": 779436189,
|
| 137 |
+
"run_shapes": [
|
| 138 |
+
"rectangle_03",
|
| 139 |
+
"polygon_3",
|
| 140 |
+
"polygon_4",
|
| 141 |
+
"star_6_r04",
|
| 142 |
+
"ellipse_050",
|
| 143 |
+
"heart",
|
| 144 |
+
"flower_2_w03",
|
| 145 |
+
"flower_5_w03",
|
| 146 |
+
"flower_6_w03",
|
| 147 |
+
"drop_15",
|
| 148 |
+
"closed_hollow_polygon_3",
|
| 149 |
+
"closed_hollow_polygon_4",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"training": {
|
| 155 |
+
"base_seed": 42,
|
| 156 |
+
"batch_size": 128,
|
| 157 |
+
"bbox_loss_inside_weight": 1.0,
|
| 158 |
+
"bbox_loss_outside_weight": 1.0,
|
| 159 |
+
"epochs": 100,
|
| 160 |
+
"lr": 0.0001,
|
| 161 |
+
"num_filters": 64,
|
| 162 |
+
"num_values": 2,
|
| 163 |
+
"num_workers": 8,
|
| 164 |
+
"seed": 779436189
|
| 165 |
+
},
|
| 166 |
+
"z_dim": 10
|
| 167 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_15shapes_zdim_010/sweep_run_config.yaml
ADDED
|
@@ -0,0 +1,142 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_index: 0
|
| 2 |
+
rerun_index: 0
|
| 3 |
+
run_name: run_000_rerun_00_15shapes_zdim_010
|
| 4 |
+
run_seed: 779436189
|
| 5 |
+
z_dim: 10
|
| 6 |
+
n_shapes: 15
|
| 7 |
+
held_out_shape: rectangle_03
|
| 8 |
+
run_shapes:
|
| 9 |
+
- rectangle_03
|
| 10 |
+
- polygon_3
|
| 11 |
+
- polygon_4
|
| 12 |
+
- star_6_r04
|
| 13 |
+
- ellipse_050
|
| 14 |
+
- heart
|
| 15 |
+
- flower_2_w03
|
| 16 |
+
- flower_5_w03
|
| 17 |
+
- flower_6_w03
|
| 18 |
+
- drop_15
|
| 19 |
+
- closed_hollow_polygon_3
|
| 20 |
+
- closed_hollow_polygon_4
|
| 21 |
+
- hollow_polygon_4
|
| 22 |
+
- hollow_polygon_5
|
| 23 |
+
- closed_hollow_ellipse_05
|
| 24 |
+
cache:
|
| 25 |
+
path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 26 |
+
available_shapes:
|
| 27 |
+
- polygon_3
|
| 28 |
+
- polygon_4
|
| 29 |
+
- polygon_5
|
| 30 |
+
- star_3_r02
|
| 31 |
+
- star_4_r04
|
| 32 |
+
- star_5_r04
|
| 33 |
+
- star_6_r04
|
| 34 |
+
- ellipse_050
|
| 35 |
+
- heart
|
| 36 |
+
- flower_2_w03
|
| 37 |
+
- flower_3_w03
|
| 38 |
+
- flower_4_w03
|
| 39 |
+
- flower_5_w03
|
| 40 |
+
- flower_6_w03
|
| 41 |
+
- rectangle_03
|
| 42 |
+
- cross_065
|
| 43 |
+
- crescent_025
|
| 44 |
+
- drop_15
|
| 45 |
+
- closed_hollow_polygon_3
|
| 46 |
+
- closed_hollow_polygon_4
|
| 47 |
+
- closed_hollow_polygon_5
|
| 48 |
+
- hollow_polygon_3
|
| 49 |
+
- hollow_polygon_4
|
| 50 |
+
- hollow_polygon_5
|
| 51 |
+
- closed_hollow_ellipse_05
|
| 52 |
+
generation_config:
|
| 53 |
+
seed: 42
|
| 54 |
+
image_size: 64
|
| 55 |
+
num_values: 2
|
| 56 |
+
output_channels: 1
|
| 57 |
+
output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 58 |
+
scale_space: renderer
|
| 59 |
+
position_space: renderer
|
| 60 |
+
shape_pool: all
|
| 61 |
+
candidate_fg_colors:
|
| 62 |
+
- - 255
|
| 63 |
+
- 255
|
| 64 |
+
- 255
|
| 65 |
+
candidate_bg_colors:
|
| 66 |
+
- - 0
|
| 67 |
+
- 0
|
| 68 |
+
- 0
|
| 69 |
+
scales:
|
| 70 |
+
- 0.2
|
| 71 |
+
- 0.24
|
| 72 |
+
- 0.28
|
| 73 |
+
- 0.32
|
| 74 |
+
- 0.36
|
| 75 |
+
- 0.4
|
| 76 |
+
orientations:
|
| 77 |
+
min: 0.0
|
| 78 |
+
max: 6.28318531
|
| 79 |
+
num: 40
|
| 80 |
+
positions_x:
|
| 81 |
+
min: 0.2
|
| 82 |
+
max: 0.8
|
| 83 |
+
num: 32
|
| 84 |
+
positions_y:
|
| 85 |
+
min: 0.2
|
| 86 |
+
max: 0.8
|
| 87 |
+
num: 32
|
| 88 |
+
train:
|
| 89 |
+
num_shapes: 25
|
| 90 |
+
num_fg_colors: 1
|
| 91 |
+
num_bg_colors: 1
|
| 92 |
+
ood:
|
| 93 |
+
num_shapes: 0
|
| 94 |
+
shapes_from: unseen
|
| 95 |
+
num_fg_colors: 0
|
| 96 |
+
fg_colors_from: unseen
|
| 97 |
+
num_bg_colors: 0
|
| 98 |
+
bg_colors_from: unseen
|
| 99 |
+
dataloader:
|
| 100 |
+
sprite_shapes:
|
| 101 |
+
- rectangle_03
|
| 102 |
+
- polygon_3
|
| 103 |
+
- polygon_4
|
| 104 |
+
- star_6_r04
|
| 105 |
+
- ellipse_050
|
| 106 |
+
- heart
|
| 107 |
+
- flower_2_w03
|
| 108 |
+
- flower_5_w03
|
| 109 |
+
- flower_6_w03
|
| 110 |
+
- drop_15
|
| 111 |
+
- closed_hollow_polygon_3
|
| 112 |
+
- closed_hollow_polygon_4
|
| 113 |
+
- hollow_polygon_4
|
| 114 |
+
- hollow_polygon_5
|
| 115 |
+
- closed_hollow_ellipse_05
|
| 116 |
+
sprite_restricted_shapes:
|
| 117 |
+
- rectangle_03
|
| 118 |
+
sprite_right_side_threshold: 0.5
|
| 119 |
+
val_fraction: 0.0
|
| 120 |
+
composition:
|
| 121 |
+
composition_op: fixint
|
| 122 |
+
command_sampler: lils
|
| 123 |
+
action_factors:
|
| 124 |
+
- shape
|
| 125 |
+
- scale
|
| 126 |
+
- orientation
|
| 127 |
+
- position_x
|
| 128 |
+
- position_y
|
| 129 |
+
ignored_factors:
|
| 130 |
+
- foreground_color
|
| 131 |
+
- background_color
|
| 132 |
+
training:
|
| 133 |
+
epochs: 100
|
| 134 |
+
batch_size: 128
|
| 135 |
+
num_workers: 8
|
| 136 |
+
num_values: 2
|
| 137 |
+
num_filters: 64
|
| 138 |
+
lr: 0.0001
|
| 139 |
+
seed: 779436189
|
| 140 |
+
base_seed: 42
|
| 141 |
+
bbox_loss_inside_weight: 1.0
|
| 142 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ff58ba8a472875933ff087481084f3cc84c97d1a6b03e4bb6452a0be0fd2732
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,229 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 4792320,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 4915200,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 178782122,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"flower_3_w03"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"flower_3_w03",
|
| 194 |
+
"polygon_3",
|
| 195 |
+
"polygon_4",
|
| 196 |
+
"star_3_r02",
|
| 197 |
+
"star_4_r04",
|
| 198 |
+
"star_5_r04",
|
| 199 |
+
"star_6_r04",
|
| 200 |
+
"ellipse_050",
|
| 201 |
+
"flower_2_w03",
|
| 202 |
+
"flower_4_w03",
|
| 203 |
+
"flower_5_w03",
|
| 204 |
+
"rectangle_03",
|
| 205 |
+
"cross_065",
|
| 206 |
+
"crescent_025",
|
| 207 |
+
"drop_15",
|
| 208 |
+
"closed_hollow_polygon_3",
|
| 209 |
+
"hollow_polygon_3",
|
| 210 |
+
"hollow_polygon_4",
|
| 211 |
+
"hollow_polygon_5",
|
| 212 |
+
"closed_hollow_ellipse_05"
|
| 213 |
+
],
|
| 214 |
+
"val_fraction": 0.0
|
| 215 |
+
},
|
| 216 |
+
"splits": {
|
| 217 |
+
"held_out_ood_size": 122880,
|
| 218 |
+
"in_domain_pool_size": 4792320,
|
| 219 |
+
"ood_validation_pool_size": 4915200,
|
| 220 |
+
"train_anchor_size": 4792319,
|
| 221 |
+
"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.",
|
| 222 |
+
"val_anchor_size": 1,
|
| 223 |
+
"val_in_domain_anchor_size": 1,
|
| 224 |
+
"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.",
|
| 225 |
+
"val_ood_anchor_size": 122880,
|
| 226 |
+
"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.",
|
| 227 |
+
"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."
|
| 228 |
+
}
|
| 229 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/events.out.tfevents.1775566879.node07.1629786.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2e3086914f2fb894a9ef355038c4e6d7d7145ee143da08b75e2c482dec5d771b
|
| 3 |
+
size 142002
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12456/hparams.yaml
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
num_filters: 64
|
| 2 |
+
z_dim: 10
|
| 3 |
+
lr: 0.0001
|
| 4 |
+
composition_op: fixint
|
| 5 |
+
n_actions: 5
|
| 6 |
+
num_channels: 1
|
| 7 |
+
num_values: 2
|
| 8 |
+
image_size: 64
|
| 9 |
+
bbox_loss_inside_weight: 1.0
|
| 10 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.json
ADDED
|
@@ -0,0 +1,177 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
|
| 91 |
+
},
|
| 92 |
+
"composition": {
|
| 93 |
+
"action_factors": [
|
| 94 |
+
"shape",
|
| 95 |
+
"scale",
|
| 96 |
+
"orientation",
|
| 97 |
+
"position_x",
|
| 98 |
+
"position_y"
|
| 99 |
+
],
|
| 100 |
+
"command_sampler": "lils",
|
| 101 |
+
"composition_op": "fixint",
|
| 102 |
+
"ignored_factors": [
|
| 103 |
+
"foreground_color",
|
| 104 |
+
"background_color"
|
| 105 |
+
]
|
| 106 |
+
},
|
| 107 |
+
"dataloader": {
|
| 108 |
+
"sprite_restricted_shapes": [
|
| 109 |
+
"flower_3_w03"
|
| 110 |
+
],
|
| 111 |
+
"sprite_right_side_threshold": 0.5,
|
| 112 |
+
"sprite_shapes": [
|
| 113 |
+
"flower_3_w03",
|
| 114 |
+
"polygon_3",
|
| 115 |
+
"polygon_4",
|
| 116 |
+
"star_3_r02",
|
| 117 |
+
"star_4_r04",
|
| 118 |
+
"star_5_r04",
|
| 119 |
+
"star_6_r04",
|
| 120 |
+
"ellipse_050",
|
| 121 |
+
"flower_2_w03",
|
| 122 |
+
"flower_4_w03",
|
| 123 |
+
"flower_5_w03",
|
| 124 |
+
"rectangle_03",
|
| 125 |
+
"cross_065",
|
| 126 |
+
"crescent_025",
|
| 127 |
+
"drop_15",
|
| 128 |
+
"closed_hollow_polygon_3",
|
| 129 |
+
"hollow_polygon_3",
|
| 130 |
+
"hollow_polygon_4",
|
| 131 |
+
"hollow_polygon_5",
|
| 132 |
+
"closed_hollow_ellipse_05"
|
| 133 |
+
],
|
| 134 |
+
"val_fraction": 0.0
|
| 135 |
+
},
|
| 136 |
+
"held_out_shape": "flower_3_w03",
|
| 137 |
+
"n_shapes": 20,
|
| 138 |
+
"rerun_index": 0,
|
| 139 |
+
"run_index": 0,
|
| 140 |
+
"run_name": "run_000_rerun_00_20shapes_zdim_010",
|
| 141 |
+
"run_seed": 178782122,
|
| 142 |
+
"run_shapes": [
|
| 143 |
+
"flower_3_w03",
|
| 144 |
+
"polygon_3",
|
| 145 |
+
"polygon_4",
|
| 146 |
+
"star_3_r02",
|
| 147 |
+
"star_4_r04",
|
| 148 |
+
"star_5_r04",
|
| 149 |
+
"star_6_r04",
|
| 150 |
+
"ellipse_050",
|
| 151 |
+
"flower_2_w03",
|
| 152 |
+
"flower_4_w03",
|
| 153 |
+
"flower_5_w03",
|
| 154 |
+
"rectangle_03",
|
| 155 |
+
"cross_065",
|
| 156 |
+
"crescent_025",
|
| 157 |
+
"drop_15",
|
| 158 |
+
"closed_hollow_polygon_3",
|
| 159 |
+
"hollow_polygon_3",
|
| 160 |
+
"hollow_polygon_4",
|
| 161 |
+
"hollow_polygon_5",
|
| 162 |
+
"closed_hollow_ellipse_05"
|
| 163 |
+
],
|
| 164 |
+
"training": {
|
| 165 |
+
"base_seed": 42,
|
| 166 |
+
"batch_size": 128,
|
| 167 |
+
"bbox_loss_inside_weight": 1.0,
|
| 168 |
+
"bbox_loss_outside_weight": 1.0,
|
| 169 |
+
"epochs": 100,
|
| 170 |
+
"lr": 0.0001,
|
| 171 |
+
"num_filters": 64,
|
| 172 |
+
"num_values": 2,
|
| 173 |
+
"num_workers": 8,
|
| 174 |
+
"seed": 178782122
|
| 175 |
+
},
|
| 176 |
+
"z_dim": 10
|
| 177 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_20shapes_zdim_010/sweep_run_config.yaml
ADDED
|
@@ -0,0 +1,152 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_index: 0
|
| 2 |
+
rerun_index: 0
|
| 3 |
+
run_name: run_000_rerun_00_20shapes_zdim_010
|
| 4 |
+
run_seed: 178782122
|
| 5 |
+
z_dim: 10
|
| 6 |
+
n_shapes: 20
|
| 7 |
+
held_out_shape: flower_3_w03
|
| 8 |
+
run_shapes:
|
| 9 |
+
- flower_3_w03
|
| 10 |
+
- polygon_3
|
| 11 |
+
- polygon_4
|
| 12 |
+
- star_3_r02
|
| 13 |
+
- star_4_r04
|
| 14 |
+
- star_5_r04
|
| 15 |
+
- star_6_r04
|
| 16 |
+
- ellipse_050
|
| 17 |
+
- flower_2_w03
|
| 18 |
+
- flower_4_w03
|
| 19 |
+
- flower_5_w03
|
| 20 |
+
- rectangle_03
|
| 21 |
+
- cross_065
|
| 22 |
+
- crescent_025
|
| 23 |
+
- drop_15
|
| 24 |
+
- closed_hollow_polygon_3
|
| 25 |
+
- hollow_polygon_3
|
| 26 |
+
- hollow_polygon_4
|
| 27 |
+
- hollow_polygon_5
|
| 28 |
+
- closed_hollow_ellipse_05
|
| 29 |
+
cache:
|
| 30 |
+
path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 31 |
+
available_shapes:
|
| 32 |
+
- polygon_3
|
| 33 |
+
- polygon_4
|
| 34 |
+
- polygon_5
|
| 35 |
+
- star_3_r02
|
| 36 |
+
- star_4_r04
|
| 37 |
+
- star_5_r04
|
| 38 |
+
- star_6_r04
|
| 39 |
+
- ellipse_050
|
| 40 |
+
- heart
|
| 41 |
+
- flower_2_w03
|
| 42 |
+
- flower_3_w03
|
| 43 |
+
- flower_4_w03
|
| 44 |
+
- flower_5_w03
|
| 45 |
+
- flower_6_w03
|
| 46 |
+
- rectangle_03
|
| 47 |
+
- cross_065
|
| 48 |
+
- crescent_025
|
| 49 |
+
- drop_15
|
| 50 |
+
- closed_hollow_polygon_3
|
| 51 |
+
- closed_hollow_polygon_4
|
| 52 |
+
- closed_hollow_polygon_5
|
| 53 |
+
- hollow_polygon_3
|
| 54 |
+
- hollow_polygon_4
|
| 55 |
+
- hollow_polygon_5
|
| 56 |
+
- closed_hollow_ellipse_05
|
| 57 |
+
generation_config:
|
| 58 |
+
seed: 42
|
| 59 |
+
image_size: 64
|
| 60 |
+
num_values: 2
|
| 61 |
+
output_channels: 1
|
| 62 |
+
output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
|
| 63 |
+
scale_space: renderer
|
| 64 |
+
position_space: renderer
|
| 65 |
+
shape_pool: all
|
| 66 |
+
candidate_fg_colors:
|
| 67 |
+
- - 255
|
| 68 |
+
- 255
|
| 69 |
+
- 255
|
| 70 |
+
candidate_bg_colors:
|
| 71 |
+
- - 0
|
| 72 |
+
- 0
|
| 73 |
+
- 0
|
| 74 |
+
scales:
|
| 75 |
+
- 0.2
|
| 76 |
+
- 0.24
|
| 77 |
+
- 0.28
|
| 78 |
+
- 0.32
|
| 79 |
+
- 0.36
|
| 80 |
+
- 0.4
|
| 81 |
+
orientations:
|
| 82 |
+
min: 0.0
|
| 83 |
+
max: 6.28318531
|
| 84 |
+
num: 40
|
| 85 |
+
positions_x:
|
| 86 |
+
min: 0.2
|
| 87 |
+
max: 0.8
|
| 88 |
+
num: 32
|
| 89 |
+
positions_y:
|
| 90 |
+
min: 0.2
|
| 91 |
+
max: 0.8
|
| 92 |
+
num: 32
|
| 93 |
+
train:
|
| 94 |
+
num_shapes: 25
|
| 95 |
+
num_fg_colors: 1
|
| 96 |
+
num_bg_colors: 1
|
| 97 |
+
ood:
|
| 98 |
+
num_shapes: 0
|
| 99 |
+
shapes_from: unseen
|
| 100 |
+
num_fg_colors: 0
|
| 101 |
+
fg_colors_from: unseen
|
| 102 |
+
num_bg_colors: 0
|
| 103 |
+
bg_colors_from: unseen
|
| 104 |
+
dataloader:
|
| 105 |
+
sprite_shapes:
|
| 106 |
+
- flower_3_w03
|
| 107 |
+
- polygon_3
|
| 108 |
+
- polygon_4
|
| 109 |
+
- star_3_r02
|
| 110 |
+
- star_4_r04
|
| 111 |
+
- star_5_r04
|
| 112 |
+
- star_6_r04
|
| 113 |
+
- ellipse_050
|
| 114 |
+
- flower_2_w03
|
| 115 |
+
- flower_4_w03
|
| 116 |
+
- flower_5_w03
|
| 117 |
+
- rectangle_03
|
| 118 |
+
- cross_065
|
| 119 |
+
- crescent_025
|
| 120 |
+
- drop_15
|
| 121 |
+
- closed_hollow_polygon_3
|
| 122 |
+
- hollow_polygon_3
|
| 123 |
+
- hollow_polygon_4
|
| 124 |
+
- hollow_polygon_5
|
| 125 |
+
- closed_hollow_ellipse_05
|
| 126 |
+
sprite_restricted_shapes:
|
| 127 |
+
- flower_3_w03
|
| 128 |
+
sprite_right_side_threshold: 0.5
|
| 129 |
+
val_fraction: 0.0
|
| 130 |
+
composition:
|
| 131 |
+
composition_op: fixint
|
| 132 |
+
command_sampler: lils
|
| 133 |
+
action_factors:
|
| 134 |
+
- shape
|
| 135 |
+
- scale
|
| 136 |
+
- orientation
|
| 137 |
+
- position_x
|
| 138 |
+
- position_y
|
| 139 |
+
ignored_factors:
|
| 140 |
+
- foreground_color
|
| 141 |
+
- background_color
|
| 142 |
+
training:
|
| 143 |
+
epochs: 100
|
| 144 |
+
batch_size: 128
|
| 145 |
+
num_workers: 8
|
| 146 |
+
num_values: 2
|
| 147 |
+
num_filters: 64
|
| 148 |
+
lr: 0.0001
|
| 149 |
+
seed: 178782122
|
| 150 |
+
base_seed: 42
|
| 151 |
+
bbox_loss_inside_weight: 1.0
|
| 152 |
+
bbox_loss_outside_weight: 1.0
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/checkpoints/best_in_domain.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1bc4d211bf2f3927bed98ec191854610d382b74526d80a77620b61c6455a064a
|
| 3 |
+
size 12959138
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/dataset_loader_config.json
ADDED
|
@@ -0,0 +1,234 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cache": {
|
| 3 |
+
"available_shapes": [
|
| 4 |
+
"polygon_3",
|
| 5 |
+
"polygon_4",
|
| 6 |
+
"polygon_5",
|
| 7 |
+
"star_3_r02",
|
| 8 |
+
"star_4_r04",
|
| 9 |
+
"star_5_r04",
|
| 10 |
+
"star_6_r04",
|
| 11 |
+
"ellipse_050",
|
| 12 |
+
"heart",
|
| 13 |
+
"flower_2_w03",
|
| 14 |
+
"flower_3_w03",
|
| 15 |
+
"flower_4_w03",
|
| 16 |
+
"flower_5_w03",
|
| 17 |
+
"flower_6_w03",
|
| 18 |
+
"rectangle_03",
|
| 19 |
+
"cross_065",
|
| 20 |
+
"crescent_025",
|
| 21 |
+
"drop_15",
|
| 22 |
+
"closed_hollow_polygon_3",
|
| 23 |
+
"closed_hollow_polygon_4",
|
| 24 |
+
"closed_hollow_polygon_5",
|
| 25 |
+
"hollow_polygon_3",
|
| 26 |
+
"hollow_polygon_4",
|
| 27 |
+
"hollow_polygon_5",
|
| 28 |
+
"closed_hollow_ellipse_05"
|
| 29 |
+
],
|
| 30 |
+
"generation_config": {
|
| 31 |
+
"candidate_bg_colors": [
|
| 32 |
+
[
|
| 33 |
+
0,
|
| 34 |
+
0,
|
| 35 |
+
0
|
| 36 |
+
]
|
| 37 |
+
],
|
| 38 |
+
"candidate_fg_colors": [
|
| 39 |
+
[
|
| 40 |
+
255,
|
| 41 |
+
255,
|
| 42 |
+
255
|
| 43 |
+
]
|
| 44 |
+
],
|
| 45 |
+
"image_size": 64,
|
| 46 |
+
"num_values": 2,
|
| 47 |
+
"ood": {
|
| 48 |
+
"bg_colors_from": "unseen",
|
| 49 |
+
"fg_colors_from": "unseen",
|
| 50 |
+
"num_bg_colors": 0,
|
| 51 |
+
"num_fg_colors": 0,
|
| 52 |
+
"num_shapes": 0,
|
| 53 |
+
"shapes_from": "unseen"
|
| 54 |
+
},
|
| 55 |
+
"orientations": {
|
| 56 |
+
"max": 6.28318531,
|
| 57 |
+
"min": 0.0,
|
| 58 |
+
"num": 40
|
| 59 |
+
},
|
| 60 |
+
"output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 61 |
+
"output_channels": 1,
|
| 62 |
+
"position_space": "renderer",
|
| 63 |
+
"positions_x": {
|
| 64 |
+
"max": 0.8,
|
| 65 |
+
"min": 0.2,
|
| 66 |
+
"num": 32
|
| 67 |
+
},
|
| 68 |
+
"positions_y": {
|
| 69 |
+
"max": 0.8,
|
| 70 |
+
"min": 0.2,
|
| 71 |
+
"num": 32
|
| 72 |
+
},
|
| 73 |
+
"scale_space": "renderer",
|
| 74 |
+
"scales": [
|
| 75 |
+
0.2,
|
| 76 |
+
0.24,
|
| 77 |
+
0.28,
|
| 78 |
+
0.32,
|
| 79 |
+
0.36,
|
| 80 |
+
0.4
|
| 81 |
+
],
|
| 82 |
+
"seed": 42,
|
| 83 |
+
"shape_pool": "all",
|
| 84 |
+
"train": {
|
| 85 |
+
"num_bg_colors": 1,
|
| 86 |
+
"num_fg_colors": 1,
|
| 87 |
+
"num_shapes": 25
|
| 88 |
+
}
|
| 89 |
+
},
|
| 90 |
+
"has_factor_structure": false,
|
| 91 |
+
"image_size": 64,
|
| 92 |
+
"num_images": 6144000,
|
| 93 |
+
"num_values": 2,
|
| 94 |
+
"output_channels": 1,
|
| 95 |
+
"path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
|
| 96 |
+
"split_config": {
|
| 97 |
+
"bg_colors_from": "unseen",
|
| 98 |
+
"fg_colors_from": "unseen",
|
| 99 |
+
"ood_bg_colors": [],
|
| 100 |
+
"ood_fg_colors": [],
|
| 101 |
+
"ood_index_range": {
|
| 102 |
+
"start": 6144000,
|
| 103 |
+
"stop": 6144000
|
| 104 |
+
},
|
| 105 |
+
"ood_shapes": [],
|
| 106 |
+
"ood_size": 0,
|
| 107 |
+
"seed": 42,
|
| 108 |
+
"shapes_from": "unseen",
|
| 109 |
+
"train_bg_colors": [
|
| 110 |
+
[
|
| 111 |
+
0,
|
| 112 |
+
0,
|
| 113 |
+
0
|
| 114 |
+
]
|
| 115 |
+
],
|
| 116 |
+
"train_fg_colors": [
|
| 117 |
+
[
|
| 118 |
+
1,
|
| 119 |
+
1,
|
| 120 |
+
1
|
| 121 |
+
]
|
| 122 |
+
],
|
| 123 |
+
"train_index_range": {
|
| 124 |
+
"start": 0,
|
| 125 |
+
"stop": 6144000
|
| 126 |
+
},
|
| 127 |
+
"train_shapes": [
|
| 128 |
+
"polygon_3",
|
| 129 |
+
"polygon_4",
|
| 130 |
+
"polygon_5",
|
| 131 |
+
"star_3_r02",
|
| 132 |
+
"star_4_r04",
|
| 133 |
+
"star_5_r04",
|
| 134 |
+
"star_6_r04",
|
| 135 |
+
"ellipse_050",
|
| 136 |
+
"heart",
|
| 137 |
+
"flower_2_w03",
|
| 138 |
+
"flower_3_w03",
|
| 139 |
+
"flower_4_w03",
|
| 140 |
+
"flower_5_w03",
|
| 141 |
+
"flower_6_w03",
|
| 142 |
+
"rectangle_03",
|
| 143 |
+
"cross_065",
|
| 144 |
+
"crescent_025",
|
| 145 |
+
"drop_15",
|
| 146 |
+
"closed_hollow_polygon_3",
|
| 147 |
+
"closed_hollow_polygon_4",
|
| 148 |
+
"closed_hollow_polygon_5",
|
| 149 |
+
"hollow_polygon_3",
|
| 150 |
+
"hollow_polygon_4",
|
| 151 |
+
"hollow_polygon_5",
|
| 152 |
+
"closed_hollow_ellipse_05"
|
| 153 |
+
],
|
| 154 |
+
"train_size": 6144000
|
| 155 |
+
}
|
| 156 |
+
},
|
| 157 |
+
"composition": {
|
| 158 |
+
"action_factors": [
|
| 159 |
+
"shape",
|
| 160 |
+
"scale",
|
| 161 |
+
"orientation",
|
| 162 |
+
"position_x",
|
| 163 |
+
"position_y"
|
| 164 |
+
],
|
| 165 |
+
"command_sampler": "lils",
|
| 166 |
+
"held_out_ood_size": 122880,
|
| 167 |
+
"ignored_factors": [
|
| 168 |
+
"foreground_color",
|
| 169 |
+
"background_color"
|
| 170 |
+
],
|
| 171 |
+
"in_domain_pool_size": 6021120,
|
| 172 |
+
"ood_excluded_from_pool": true,
|
| 173 |
+
"ood_excluded_from_training_pool": true,
|
| 174 |
+
"ood_validation_pool_size": 6144000,
|
| 175 |
+
"ood_validation_uses_combined_pool": true,
|
| 176 |
+
"split_source": "restricted_shape_holdout",
|
| 177 |
+
"task_factors": [
|
| 178 |
+
"shape",
|
| 179 |
+
"scale",
|
| 180 |
+
"orientation",
|
| 181 |
+
"position_x",
|
| 182 |
+
"position_y"
|
| 183 |
+
]
|
| 184 |
+
},
|
| 185 |
+
"dataset_name": "sprites_large_composition",
|
| 186 |
+
"loader_filter": {
|
| 187 |
+
"seed": 1192396634,
|
| 188 |
+
"sprite_restricted_shapes": [
|
| 189 |
+
"ellipse_050"
|
| 190 |
+
],
|
| 191 |
+
"sprite_right_side_threshold": 0.5,
|
| 192 |
+
"sprite_shapes": [
|
| 193 |
+
"ellipse_050",
|
| 194 |
+
"polygon_3",
|
| 195 |
+
"polygon_4",
|
| 196 |
+
"polygon_5",
|
| 197 |
+
"star_3_r02",
|
| 198 |
+
"star_4_r04",
|
| 199 |
+
"star_5_r04",
|
| 200 |
+
"star_6_r04",
|
| 201 |
+
"heart",
|
| 202 |
+
"flower_2_w03",
|
| 203 |
+
"flower_3_w03",
|
| 204 |
+
"flower_4_w03",
|
| 205 |
+
"flower_5_w03",
|
| 206 |
+
"flower_6_w03",
|
| 207 |
+
"rectangle_03",
|
| 208 |
+
"cross_065",
|
| 209 |
+
"crescent_025",
|
| 210 |
+
"drop_15",
|
| 211 |
+
"closed_hollow_polygon_3",
|
| 212 |
+
"closed_hollow_polygon_4",
|
| 213 |
+
"closed_hollow_polygon_5",
|
| 214 |
+
"hollow_polygon_3",
|
| 215 |
+
"hollow_polygon_4",
|
| 216 |
+
"hollow_polygon_5",
|
| 217 |
+
"closed_hollow_ellipse_05"
|
| 218 |
+
],
|
| 219 |
+
"val_fraction": 0.0
|
| 220 |
+
},
|
| 221 |
+
"splits": {
|
| 222 |
+
"held_out_ood_size": 122880,
|
| 223 |
+
"in_domain_pool_size": 6021120,
|
| 224 |
+
"ood_validation_pool_size": 6144000,
|
| 225 |
+
"train_anchor_size": 6021119,
|
| 226 |
+
"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.",
|
| 227 |
+
"val_anchor_size": 1,
|
| 228 |
+
"val_in_domain_anchor_size": 1,
|
| 229 |
+
"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.",
|
| 230 |
+
"val_ood_anchor_size": 122880,
|
| 231 |
+
"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.",
|
| 232 |
+
"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."
|
| 233 |
+
}
|
| 234 |
+
}
|
CompVAE_logs/dsprites_original_large/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12457/events.out.tfevents.1775566899.node07.1629937.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ca1e3edceef3c389821126aac829eea9d5b81ec42679cdcf6a71da58781ccc1a
|
| 3 |
+
size 131623
|