Sync log files from VAE_logs and CompVAE_logs (1/2)

CompVAE_logs/dsprites_original_large/composition_sweep_plan.json

@@ -0,0 +1,183 @@
+{
+  "available_shapes": [
+    "polygon_3",
+    "polygon_4",
+    "polygon_5",
+    "star_3_r02",
+    "star_4_r04",
+    "star_5_r04",
+    "star_6_r04",
+    "ellipse_050",
+    "heart",
+    "flower_2_w03",
+    "flower_3_w03",
+    "flower_4_w03",
+    "flower_5_w03",
+    "flower_6_w03",
+    "rectangle_03",
+    "cross_065",
+    "crescent_025",
+    "drop_15",
+    "closed_hollow_polygon_3",
+    "closed_hollow_polygon_4",
+    "closed_hollow_polygon_5",
+    "hollow_polygon_3",
+    "hollow_polygon_4",
+    "hollow_polygon_5",
+    "closed_hollow_ellipse_05"
+  ],
+  "base_generation_config": {
+    "candidate_bg_colors": [
+      [
+        0,
+        0,
+        0
+      ]
+    ],
+    "candidate_fg_colors": [
+      [
+        255,
+        255,
+        255
+      ]
+    ],
+    "image_size": 64,
+    "num_values": 2,
+    "ood": {
+      "bg_colors_from": "unseen",
+      "fg_colors_from": "unseen",
+      "num_bg_colors": 0,
+      "num_fg_colors": 0,
+      "num_shapes": 0,
+      "shapes_from": "unseen"
+    },
+    "orientations": {
+      "max": 6.28318531,
+      "min": 0.0,
+      "num": 40
+    },
+    "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+    "output_channels": 1,
+    "position_space": "renderer",
+    "positions_x": {
+      "max": 0.8,
+      "min": 0.2,
+      "num": 32
+    },
+    "positions_y": {
+      "max": 0.8,
+      "min": 0.2,
+      "num": 32
+    },
+    "scale_space": "renderer",
+    "scales": [
+      0.2,
+      0.24,
+      0.28,
+      0.32,
+      0.36,
+      0.4
+    ],
+    "seed": 42,
+    "shape_pool": "all",
+    "train": {
+      "num_bg_colors": 1,
+      "num_fg_colors": 1,
+      "num_shapes": 25
+    }
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "composition_op": "fixint",
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ]
+  },
+  "n_shapes_sweep": [
+    3
+  ],
+  "num_reruns": 3,
+  "rerun_offset": 0,
+  "runs": [
+    {
+      "held_out_shape": "polygon_3",
+      "n_shapes": 3,
+      "rerun_index": 0,
+      "run_index": 0,
+      "run_seed": 2548879074,
+      "run_shapes": [
+        "polygon_3",
+        "polygon_5",
+        "star_5_r04"
+      ],
+      "z_dim": 10
+    },
+    {
+      "held_out_shape": "flower_6_w03",
+      "n_shapes": 3,
+      "rerun_index": 1,
+      "run_index": 1,
+      "run_seed": 2389822288,
+      "run_shapes": [
+        "flower_6_w03",
+        "flower_5_w03",
+        "cross_065"
+      ],
+      "z_dim": 10
+    },
+    {
+      "held_out_shape": "closed_hollow_polygon_4",
+      "n_shapes": 3,
+      "rerun_index": 2,
+      "run_index": 2,
+      "run_seed": 1247230849,
+      "run_shapes": [
+        "closed_hollow_polygon_4",
+        "flower_3_w03",
+        "hollow_polygon_5"
+      ],
+      "z_dim": 10
+    }
+  ],
+  "shape_pool": [
+    "polygon_3",
+    "polygon_4",
+    "polygon_5",
+    "star_3_r02",
+    "star_4_r04",
+    "star_5_r04",
+    "star_6_r04",
+    "ellipse_050",
+    "heart",
+    "flower_2_w03",
+    "flower_3_w03",
+    "flower_4_w03",
+    "flower_5_w03",
+    "flower_6_w03",
+    "rectangle_03",
+    "cross_065",
+    "crescent_025",
+    "drop_15",
+    "closed_hollow_polygon_3",
+    "closed_hollow_polygon_4",
+    "closed_hollow_polygon_5",
+    "hollow_polygon_3",
+    "hollow_polygon_4",
+    "hollow_polygon_5",
+    "closed_hollow_ellipse_05"
+  ],
+  "sprite_right_side_threshold": 0.5,
+  "sprites_large_cache": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+  "val_fraction": 0.0,
+  "z_dims": [
+    10
+  ]
+}

CompVAE_logs/dsprites_original_large/composition_sweep_plan.yaml

@@ -0,0 +1,151 @@
+sprites_large_cache: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+available_shapes:
+- polygon_3
+- polygon_4
+- polygon_5
+- star_3_r02
+- star_4_r04
+- star_5_r04
+- star_6_r04
+- ellipse_050
+- heart
+- flower_2_w03
+- flower_3_w03
+- flower_4_w03
+- flower_5_w03
+- flower_6_w03
+- rectangle_03
+- cross_065
+- crescent_025
+- drop_15
+- closed_hollow_polygon_3
+- closed_hollow_polygon_4
+- closed_hollow_polygon_5
+- hollow_polygon_3
+- hollow_polygon_4
+- hollow_polygon_5
+- closed_hollow_ellipse_05
+shape_pool:
+- polygon_3
+- polygon_4
+- polygon_5
+- star_3_r02
+- star_4_r04
+- star_5_r04
+- star_6_r04
+- ellipse_050
+- heart
+- flower_2_w03
+- flower_3_w03
+- flower_4_w03
+- flower_5_w03
+- flower_6_w03
+- rectangle_03
+- cross_065
+- crescent_025
+- drop_15
+- closed_hollow_polygon_3
+- closed_hollow_polygon_4
+- closed_hollow_polygon_5
+- hollow_polygon_3
+- hollow_polygon_4
+- hollow_polygon_5
+- closed_hollow_ellipse_05
+num_reruns: 3
+rerun_offset: 0
+sprite_right_side_threshold: 0.5
+val_fraction: 0.0
+n_shapes_sweep:
+- 3
+z_dims:
+- 10
+base_generation_config:
+  seed: 42
+  image_size: 64
+  num_values: 2
+  output_channels: 1
+  output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+  scale_space: renderer
+  position_space: renderer
+  shape_pool: all
+  candidate_fg_colors:
+  - - 255
+    - 255
+    - 255
+  candidate_bg_colors:
+  - - 0
+    - 0
+    - 0
+  scales:
+  - 0.2
+  - 0.24
+  - 0.28
+  - 0.32
+  - 0.36
+  - 0.4
+  orientations:
+    min: 0.0
+    max: 6.28318531
+    num: 40
+  positions_x:
+    min: 0.2
+    max: 0.8
+    num: 32
+  positions_y:
+    min: 0.2
+    max: 0.8
+    num: 32
+  train:
+    num_shapes: 25
+    num_fg_colors: 1
+    num_bg_colors: 1
+  ood:
+    num_shapes: 0
+    shapes_from: unseen
+    num_fg_colors: 0
+    fg_colors_from: unseen
+    num_bg_colors: 0
+    bg_colors_from: unseen
+composition:
+  composition_op: fixint
+  command_sampler: lils
+  action_factors:
+  - shape
+  - scale
+  - orientation
+  - position_x
+  - position_y
+  ignored_factors:
+  - foreground_color
+  - background_color
+runs:
+- run_index: 0
+  rerun_index: 0
+  run_seed: 2548879074
+  n_shapes: 3
+  z_dim: 10
+  held_out_shape: polygon_3
+  run_shapes:
+  - polygon_3
+  - polygon_5
+  - star_5_r04
+- run_index: 1
+  rerun_index: 1
+  run_seed: 2389822288
+  n_shapes: 3
+  z_dim: 10
+  held_out_shape: flower_6_w03
+  run_shapes:
+  - flower_6_w03
+  - flower_5_w03
+  - cross_065
+- run_index: 2
+  rerun_index: 2
+  run_seed: 1247230849
+  n_shapes: 3
+  z_dim: 10
+  held_out_shape: closed_hollow_polygon_4
+  run_shapes:
+  - closed_hollow_polygon_4
+  - flower_3_w03
+  - hollow_polygon_5

CompVAE_logs/dsprites_original_large/composition_sweep_summary.json

@@ -0,0 +1,62 @@
+[
+  {
+    "checkpoint_paths": {
+      "best_in_domain": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt"
+    },
+    "held_out_shape": "polygon_3",
+    "n_shapes": 3,
+    "rerun_index": 0,
+    "run_config_json_path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json",
+    "run_config_yaml_path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml",
+    "run_index": 0,
+    "run_name": "run_000_rerun_00_03shapes_zdim_010",
+    "run_seed": 2548879074,
+    "run_shapes": [
+      "polygon_3",
+      "polygon_5",
+      "star_5_r04"
+    ],
+    "version_dir": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439",
+    "z_dim": 10
+  },
+  {
+    "checkpoint_paths": {
+      "best_in_domain": "CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt"
+    },
+    "held_out_shape": "flower_6_w03",
+    "n_shapes": 3,
+    "rerun_index": 1,
+    "run_config_json_path": "CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/sweep_run_config.json",
+    "run_config_yaml_path": "CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/sweep_run_config.yaml",
+    "run_index": 1,
+    "run_name": "run_001_rerun_01_03shapes_zdim_010",
+    "run_seed": 2389822288,
+    "run_shapes": [
+      "flower_6_w03",
+      "flower_5_w03",
+      "cross_065"
+    ],
+    "version_dir": "CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439",
+    "z_dim": 10
+  },
+  {
+    "checkpoint_paths": {
+      "best_in_domain": "CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt"
+    },
+    "held_out_shape": "closed_hollow_polygon_4",
+    "n_shapes": 3,
+    "rerun_index": 2,
+    "run_config_json_path": "CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/sweep_run_config.json",
+    "run_config_yaml_path": "CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/sweep_run_config.yaml",
+    "run_index": 2,
+    "run_name": "run_002_rerun_02_03shapes_zdim_010",
+    "run_seed": 1247230849,
+    "run_shapes": [
+      "closed_hollow_polygon_4",
+      "flower_3_w03",
+      "hollow_polygon_5"
+    ],
+    "version_dir": "CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439",
+    "z_dim": 10
+  }
+]

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/checkpoint_summary.json

@@ -0,0 +1,10 @@
+{
+  "checkpoints": {
+    "best_in_domain": {
+      "monitor": "val_in_domain_bpd",
+      "path": "CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt",
+      "score": 0.03314736485481262
+    }
+  },
+  "default_checkpoint": "best_in_domain"
+}

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3108fda1a0e784cb65606c07dfd940004f558f5eaa889ab6c422ff2cecce6d03
+size 12959407

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/dataset_loader_config.json

@@ -0,0 +1,212 @@
+{
+  "cache": {
+    "available_shapes": [
+      "polygon_3",
+      "polygon_4",
+      "polygon_5",
+      "star_3_r02",
+      "star_4_r04",
+      "star_5_r04",
+      "star_6_r04",
+      "ellipse_050",
+      "heart",
+      "flower_2_w03",
+      "flower_3_w03",
+      "flower_4_w03",
+      "flower_5_w03",
+      "flower_6_w03",
+      "rectangle_03",
+      "cross_065",
+      "crescent_025",
+      "drop_15",
+      "closed_hollow_polygon_3",
+      "closed_hollow_polygon_4",
+      "closed_hollow_polygon_5",
+      "hollow_polygon_3",
+      "hollow_polygon_4",
+      "hollow_polygon_5",
+      "closed_hollow_ellipse_05"
+    ],
+    "generation_config": {
+      "candidate_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "candidate_fg_colors": [
+        [
+          255,
+          255,
+          255
+        ]
+      ],
+      "image_size": 64,
+      "num_values": 2,
+      "ood": {
+        "bg_colors_from": "unseen",
+        "fg_colors_from": "unseen",
+        "num_bg_colors": 0,
+        "num_fg_colors": 0,
+        "num_shapes": 0,
+        "shapes_from": "unseen"
+      },
+      "orientations": {
+        "max": 6.28318531,
+        "min": 0.0,
+        "num": 40
+      },
+      "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+      "output_channels": 1,
+      "position_space": "renderer",
+      "positions_x": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "positions_y": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "scale_space": "renderer",
+      "scales": [
+        0.2,
+        0.24,
+        0.28,
+        0.32,
+        0.36,
+        0.4
+      ],
+      "seed": 42,
+      "shape_pool": "all",
+      "train": {
+        "num_bg_colors": 1,
+        "num_fg_colors": 1,
+        "num_shapes": 25
+      }
+    },
+    "has_factor_structure": false,
+    "image_size": 64,
+    "num_images": 6144000,
+    "num_values": 2,
+    "output_channels": 1,
+    "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+    "split_config": {
+      "bg_colors_from": "unseen",
+      "fg_colors_from": "unseen",
+      "ood_bg_colors": [],
+      "ood_fg_colors": [],
+      "ood_index_range": {
+        "start": 6144000,
+        "stop": 6144000
+      },
+      "ood_shapes": [],
+      "ood_size": 0,
+      "seed": 42,
+      "shapes_from": "unseen",
+      "train_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "train_fg_colors": [
+        [
+          1,
+          1,
+          1
+        ]
+      ],
+      "train_index_range": {
+        "start": 0,
+        "stop": 6144000
+      },
+      "train_shapes": [
+        "polygon_3",
+        "polygon_4",
+        "polygon_5",
+        "star_3_r02",
+        "star_4_r04",
+        "star_5_r04",
+        "star_6_r04",
+        "ellipse_050",
+        "heart",
+        "flower_2_w03",
+        "flower_3_w03",
+        "flower_4_w03",
+        "flower_5_w03",
+        "flower_6_w03",
+        "rectangle_03",
+        "cross_065",
+        "crescent_025",
+        "drop_15",
+        "closed_hollow_polygon_3",
+        "closed_hollow_polygon_4",
+        "closed_hollow_polygon_5",
+        "hollow_polygon_3",
+        "hollow_polygon_4",
+        "hollow_polygon_5",
+        "closed_hollow_ellipse_05"
+      ],
+      "train_size": 6144000
+    }
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "held_out_ood_size": 0,
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ],
+    "in_domain_pool_size": 737280,
+    "ood_excluded_from_pool": true,
+    "ood_excluded_from_training_pool": true,
+    "ood_validation_pool_size": 737280,
+    "ood_validation_uses_combined_pool": false,
+    "split_source": "cache_split",
+    "task_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ]
+  },
+  "dataset_name": "sprites_large_composition",
+  "loader_filter": {
+    "seed": 2548879074,
+    "sprite_restricted_shapes": [
+      "polygon_3"
+    ],
+    "sprite_right_side_threshold": 0.5,
+    "sprite_shapes": [
+      "polygon_3",
+      "polygon_5",
+      "star_5_r04"
+    ],
+    "val_fraction": 0.0
+  },
+  "splits": {
+    "held_out_ood_size": 0,
+    "in_domain_pool_size": 737280,
+    "ood_validation_pool_size": 737280,
+    "train_anchor_size": 737280,
+    "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.",
+    "val_anchor_size": 737280,
+    "val_in_domain_anchor_size": 737280,
+    "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.",
+    "val_ood_anchor_size": 0,
+    "val_ood_rule": null,
+    "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."
+  }
+}

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/events.out.tfevents.1775553152.node04.1239199.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fa1d67b2a165071572ab7e0dd910970786170ffa7c260a40563e048a2b8f140
+size 43551

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12439/hparams.yaml

@@ -0,0 +1,10 @@
+num_filters: 64
+z_dim: 10
+lr: 0.0001
+composition_op: fixint
+n_actions: 5
+num_channels: 1
+num_values: 2
+image_size: 64
+bbox_loss_inside_weight: 1.0
+bbox_loss_outside_weight: 1.0

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.json

@@ -0,0 +1,143 @@
+{
+  "cache": {
+    "available_shapes": [
+      "polygon_3",
+      "polygon_4",
+      "polygon_5",
+      "star_3_r02",
+      "star_4_r04",
+      "star_5_r04",
+      "star_6_r04",
+      "ellipse_050",
+      "heart",
+      "flower_2_w03",
+      "flower_3_w03",
+      "flower_4_w03",
+      "flower_5_w03",
+      "flower_6_w03",
+      "rectangle_03",
+      "cross_065",
+      "crescent_025",
+      "drop_15",
+      "closed_hollow_polygon_3",
+      "closed_hollow_polygon_4",
+      "closed_hollow_polygon_5",
+      "hollow_polygon_3",
+      "hollow_polygon_4",
+      "hollow_polygon_5",
+      "closed_hollow_ellipse_05"
+    ],
+    "generation_config": {
+      "candidate_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "candidate_fg_colors": [
+        [
+          255,
+          255,
+          255
+        ]
+      ],
+      "image_size": 64,
+      "num_values": 2,
+      "ood": {
+        "bg_colors_from": "unseen",
+        "fg_colors_from": "unseen",
+        "num_bg_colors": 0,
+        "num_fg_colors": 0,
+        "num_shapes": 0,
+        "shapes_from": "unseen"
+      },
+      "orientations": {
+        "max": 6.28318531,
+        "min": 0.0,
+        "num": 40
+      },
+      "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+      "output_channels": 1,
+      "position_space": "renderer",
+      "positions_x": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "positions_y": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "scale_space": "renderer",
+      "scales": [
+        0.2,
+        0.24,
+        0.28,
+        0.32,
+        0.36,
+        0.4
+      ],
+      "seed": 42,
+      "shape_pool": "all",
+      "train": {
+        "num_bg_colors": 1,
+        "num_fg_colors": 1,
+        "num_shapes": 25
+      }
+    },
+    "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "composition_op": "fixint",
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ]
+  },
+  "dataloader": {
+    "sprite_restricted_shapes": [
+      "polygon_3"
+    ],
+    "sprite_right_side_threshold": 0.5,
+    "sprite_shapes": [
+      "polygon_3",
+      "polygon_5",
+      "star_5_r04"
+    ],
+    "val_fraction": 0.0
+  },
+  "held_out_shape": "polygon_3",
+  "n_shapes": 3,
+  "rerun_index": 0,
+  "run_index": 0,
+  "run_name": "run_000_rerun_00_03shapes_zdim_010",
+  "run_seed": 2548879074,
+  "run_shapes": [
+    "polygon_3",
+    "polygon_5",
+    "star_5_r04"
+  ],
+  "training": {
+    "base_seed": 42,
+    "batch_size": 64,
+    "bbox_loss_inside_weight": 1.0,
+    "bbox_loss_outside_weight": 1.0,
+    "epochs": 1,
+    "lr": 0.0001,
+    "num_filters": 64,
+    "num_values": 2,
+    "num_workers": 8,
+    "seed": 2548879074
+  },
+  "z_dim": 10
+}

CompVAE_logs/dsprites_original_large/run_000_rerun_00_03shapes_zdim_010/sweep_run_config.yaml

@@ -0,0 +1,118 @@
+run_index: 0
+rerun_index: 0
+run_name: run_000_rerun_00_03shapes_zdim_010
+run_seed: 2548879074
+z_dim: 10
+n_shapes: 3
+held_out_shape: polygon_3
+run_shapes:
+- polygon_3
+- polygon_5
+- star_5_r04
+cache:
+  path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+  available_shapes:
+  - polygon_3
+  - polygon_4
+  - polygon_5
+  - star_3_r02
+  - star_4_r04
+  - star_5_r04
+  - star_6_r04
+  - ellipse_050
+  - heart
+  - flower_2_w03
+  - flower_3_w03
+  - flower_4_w03
+  - flower_5_w03
+  - flower_6_w03
+  - rectangle_03
+  - cross_065
+  - crescent_025
+  - drop_15
+  - closed_hollow_polygon_3
+  - closed_hollow_polygon_4
+  - closed_hollow_polygon_5
+  - hollow_polygon_3
+  - hollow_polygon_4
+  - hollow_polygon_5
+  - closed_hollow_ellipse_05
+  generation_config:
+    seed: 42
+    image_size: 64
+    num_values: 2
+    output_channels: 1
+    output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+    scale_space: renderer
+    position_space: renderer
+    shape_pool: all
+    candidate_fg_colors:
+    - - 255
+      - 255
+      - 255
+    candidate_bg_colors:
+    - - 0
+      - 0
+      - 0
+    scales:
+    - 0.2
+    - 0.24
+    - 0.28
+    - 0.32
+    - 0.36
+    - 0.4
+    orientations:
+      min: 0.0
+      max: 6.28318531
+      num: 40
+    positions_x:
+      min: 0.2
+      max: 0.8
+      num: 32
+    positions_y:
+      min: 0.2
+      max: 0.8
+      num: 32
+    train:
+      num_shapes: 25
+      num_fg_colors: 1
+      num_bg_colors: 1
+    ood:
+      num_shapes: 0
+      shapes_from: unseen
+      num_fg_colors: 0
+      fg_colors_from: unseen
+      num_bg_colors: 0
+      bg_colors_from: unseen
+dataloader:
+  sprite_shapes:
+  - polygon_3
+  - polygon_5
+  - star_5_r04
+  sprite_restricted_shapes:
+  - polygon_3
+  sprite_right_side_threshold: 0.5
+  val_fraction: 0.0
+composition:
+  composition_op: fixint
+  command_sampler: lils
+  action_factors:
+  - shape
+  - scale
+  - orientation
+  - position_x
+  - position_y
+  ignored_factors:
+  - foreground_color
+  - background_color
+training:
+  epochs: 1
+  batch_size: 64
+  num_workers: 8
+  num_values: 2
+  num_filters: 64
+  lr: 0.0001
+  seed: 2548879074
+  base_seed: 42
+  bbox_loss_inside_weight: 1.0
+  bbox_loss_outside_weight: 1.0

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/checkpoint_summary.json

@@ -0,0 +1,10 @@
+{
+  "checkpoints": {
+    "best_in_domain": {
+      "monitor": "val_in_domain_bpd",
+      "path": "CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt",
+      "score": 0.05681132897734642
+    }
+  },
+  "default_checkpoint": "best_in_domain"
+}

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f0483ec1130202ae177a0e63bf923e7eea8bffb5e812feae1a1bf3530cb846d
+size 12959407

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/dataset_loader_config.json

@@ -0,0 +1,212 @@
+{
+  "cache": {
+    "available_shapes": [
+      "polygon_3",
+      "polygon_4",
+      "polygon_5",
+      "star_3_r02",
+      "star_4_r04",
+      "star_5_r04",
+      "star_6_r04",
+      "ellipse_050",
+      "heart",
+      "flower_2_w03",
+      "flower_3_w03",
+      "flower_4_w03",
+      "flower_5_w03",
+      "flower_6_w03",
+      "rectangle_03",
+      "cross_065",
+      "crescent_025",
+      "drop_15",
+      "closed_hollow_polygon_3",
+      "closed_hollow_polygon_4",
+      "closed_hollow_polygon_5",
+      "hollow_polygon_3",
+      "hollow_polygon_4",
+      "hollow_polygon_5",
+      "closed_hollow_ellipse_05"
+    ],
+    "generation_config": {
+      "candidate_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "candidate_fg_colors": [
+        [
+          255,
+          255,
+          255
+        ]
+      ],
+      "image_size": 64,
+      "num_values": 2,
+      "ood": {
+        "bg_colors_from": "unseen",
+        "fg_colors_from": "unseen",
+        "num_bg_colors": 0,
+        "num_fg_colors": 0,
+        "num_shapes": 0,
+        "shapes_from": "unseen"
+      },
+      "orientations": {
+        "max": 6.28318531,
+        "min": 0.0,
+        "num": 40
+      },
+      "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+      "output_channels": 1,
+      "position_space": "renderer",
+      "positions_x": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "positions_y": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "scale_space": "renderer",
+      "scales": [
+        0.2,
+        0.24,
+        0.28,
+        0.32,
+        0.36,
+        0.4
+      ],
+      "seed": 42,
+      "shape_pool": "all",
+      "train": {
+        "num_bg_colors": 1,
+        "num_fg_colors": 1,
+        "num_shapes": 25
+      }
+    },
+    "has_factor_structure": false,
+    "image_size": 64,
+    "num_images": 6144000,
+    "num_values": 2,
+    "output_channels": 1,
+    "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+    "split_config": {
+      "bg_colors_from": "unseen",
+      "fg_colors_from": "unseen",
+      "ood_bg_colors": [],
+      "ood_fg_colors": [],
+      "ood_index_range": {
+        "start": 6144000,
+        "stop": 6144000
+      },
+      "ood_shapes": [],
+      "ood_size": 0,
+      "seed": 42,
+      "shapes_from": "unseen",
+      "train_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "train_fg_colors": [
+        [
+          1,
+          1,
+          1
+        ]
+      ],
+      "train_index_range": {
+        "start": 0,
+        "stop": 6144000
+      },
+      "train_shapes": [
+        "polygon_3",
+        "polygon_4",
+        "polygon_5",
+        "star_3_r02",
+        "star_4_r04",
+        "star_5_r04",
+        "star_6_r04",
+        "ellipse_050",
+        "heart",
+        "flower_2_w03",
+        "flower_3_w03",
+        "flower_4_w03",
+        "flower_5_w03",
+        "flower_6_w03",
+        "rectangle_03",
+        "cross_065",
+        "crescent_025",
+        "drop_15",
+        "closed_hollow_polygon_3",
+        "closed_hollow_polygon_4",
+        "closed_hollow_polygon_5",
+        "hollow_polygon_3",
+        "hollow_polygon_4",
+        "hollow_polygon_5",
+        "closed_hollow_ellipse_05"
+      ],
+      "train_size": 6144000
+    }
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "held_out_ood_size": 0,
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ],
+    "in_domain_pool_size": 737280,
+    "ood_excluded_from_pool": true,
+    "ood_excluded_from_training_pool": true,
+    "ood_validation_pool_size": 737280,
+    "ood_validation_uses_combined_pool": false,
+    "split_source": "cache_split",
+    "task_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ]
+  },
+  "dataset_name": "sprites_large_composition",
+  "loader_filter": {
+    "seed": 2389822288,
+    "sprite_restricted_shapes": [
+      "flower_6_w03"
+    ],
+    "sprite_right_side_threshold": 0.5,
+    "sprite_shapes": [
+      "flower_6_w03",
+      "flower_5_w03",
+      "cross_065"
+    ],
+    "val_fraction": 0.0
+  },
+  "splits": {
+    "held_out_ood_size": 0,
+    "in_domain_pool_size": 737280,
+    "ood_validation_pool_size": 737280,
+    "train_anchor_size": 737280,
+    "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.",
+    "val_anchor_size": 737280,
+    "val_in_domain_anchor_size": 737280,
+    "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.",
+    "val_ood_anchor_size": 0,
+    "val_ood_rule": null,
+    "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."
+  }
+}

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/events.out.tfevents.1775554250.node04.1239199.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89f5f8fccb3ae92f76867bf3ca6879c5f629abd1dddec50115f12121d5ee0877
+size 44801

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12439/hparams.yaml

@@ -0,0 +1,10 @@
+num_filters: 64
+z_dim: 10
+lr: 0.0001
+composition_op: fixint
+n_actions: 5
+num_channels: 1
+num_values: 2
+image_size: 64
+bbox_loss_inside_weight: 1.0
+bbox_loss_outside_weight: 1.0

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/sweep_run_config.json

@@ -0,0 +1,143 @@
+{
+  "cache": {
+    "available_shapes": [
+      "polygon_3",
+      "polygon_4",
+      "polygon_5",
+      "star_3_r02",
+      "star_4_r04",
+      "star_5_r04",
+      "star_6_r04",
+      "ellipse_050",
+      "heart",
+      "flower_2_w03",
+      "flower_3_w03",
+      "flower_4_w03",
+      "flower_5_w03",
+      "flower_6_w03",
+      "rectangle_03",
+      "cross_065",
+      "crescent_025",
+      "drop_15",
+      "closed_hollow_polygon_3",
+      "closed_hollow_polygon_4",
+      "closed_hollow_polygon_5",
+      "hollow_polygon_3",
+      "hollow_polygon_4",
+      "hollow_polygon_5",
+      "closed_hollow_ellipse_05"
+    ],
+    "generation_config": {
+      "candidate_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "candidate_fg_colors": [
+        [
+          255,
+          255,
+          255
+        ]
+      ],
+      "image_size": 64,
+      "num_values": 2,
+      "ood": {
+        "bg_colors_from": "unseen",
+        "fg_colors_from": "unseen",
+        "num_bg_colors": 0,
+        "num_fg_colors": 0,
+        "num_shapes": 0,
+        "shapes_from": "unseen"
+      },
+      "orientations": {
+        "max": 6.28318531,
+        "min": 0.0,
+        "num": 40
+      },
+      "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+      "output_channels": 1,
+      "position_space": "renderer",
+      "positions_x": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "positions_y": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "scale_space": "renderer",
+      "scales": [
+        0.2,
+        0.24,
+        0.28,
+        0.32,
+        0.36,
+        0.4
+      ],
+      "seed": 42,
+      "shape_pool": "all",
+      "train": {
+        "num_bg_colors": 1,
+        "num_fg_colors": 1,
+        "num_shapes": 25
+      }
+    },
+    "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "composition_op": "fixint",
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ]
+  },
+  "dataloader": {
+    "sprite_restricted_shapes": [
+      "flower_6_w03"
+    ],
+    "sprite_right_side_threshold": 0.5,
+    "sprite_shapes": [
+      "flower_6_w03",
+      "flower_5_w03",
+      "cross_065"
+    ],
+    "val_fraction": 0.0
+  },
+  "held_out_shape": "flower_6_w03",
+  "n_shapes": 3,
+  "rerun_index": 1,
+  "run_index": 1,
+  "run_name": "run_001_rerun_01_03shapes_zdim_010",
+  "run_seed": 2389822288,
+  "run_shapes": [
+    "flower_6_w03",
+    "flower_5_w03",
+    "cross_065"
+  ],
+  "training": {
+    "base_seed": 42,
+    "batch_size": 64,
+    "bbox_loss_inside_weight": 1.0,
+    "bbox_loss_outside_weight": 1.0,
+    "epochs": 1,
+    "lr": 0.0001,
+    "num_filters": 64,
+    "num_values": 2,
+    "num_workers": 8,
+    "seed": 2389822288
+  },
+  "z_dim": 10
+}

CompVAE_logs/dsprites_original_large/run_001_rerun_01_03shapes_zdim_010/sweep_run_config.yaml

@@ -0,0 +1,118 @@
+run_index: 1
+rerun_index: 1
+run_name: run_001_rerun_01_03shapes_zdim_010
+run_seed: 2389822288
+z_dim: 10
+n_shapes: 3
+held_out_shape: flower_6_w03
+run_shapes:
+- flower_6_w03
+- flower_5_w03
+- cross_065
+cache:
+  path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+  available_shapes:
+  - polygon_3
+  - polygon_4
+  - polygon_5
+  - star_3_r02
+  - star_4_r04
+  - star_5_r04
+  - star_6_r04
+  - ellipse_050
+  - heart
+  - flower_2_w03
+  - flower_3_w03
+  - flower_4_w03
+  - flower_5_w03
+  - flower_6_w03
+  - rectangle_03
+  - cross_065
+  - crescent_025
+  - drop_15
+  - closed_hollow_polygon_3
+  - closed_hollow_polygon_4
+  - closed_hollow_polygon_5
+  - hollow_polygon_3
+  - hollow_polygon_4
+  - hollow_polygon_5
+  - closed_hollow_ellipse_05
+  generation_config:
+    seed: 42
+    image_size: 64
+    num_values: 2
+    output_channels: 1
+    output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+    scale_space: renderer
+    position_space: renderer
+    shape_pool: all
+    candidate_fg_colors:
+    - - 255
+      - 255
+      - 255
+    candidate_bg_colors:
+    - - 0
+      - 0
+      - 0
+    scales:
+    - 0.2
+    - 0.24
+    - 0.28
+    - 0.32
+    - 0.36
+    - 0.4
+    orientations:
+      min: 0.0
+      max: 6.28318531
+      num: 40
+    positions_x:
+      min: 0.2
+      max: 0.8
+      num: 32
+    positions_y:
+      min: 0.2
+      max: 0.8
+      num: 32
+    train:
+      num_shapes: 25
+      num_fg_colors: 1
+      num_bg_colors: 1
+    ood:
+      num_shapes: 0
+      shapes_from: unseen
+      num_fg_colors: 0
+      fg_colors_from: unseen
+      num_bg_colors: 0
+      bg_colors_from: unseen
+dataloader:
+  sprite_shapes:
+  - flower_6_w03
+  - flower_5_w03
+  - cross_065
+  sprite_restricted_shapes:
+  - flower_6_w03
+  sprite_right_side_threshold: 0.5
+  val_fraction: 0.0
+composition:
+  composition_op: fixint
+  command_sampler: lils
+  action_factors:
+  - shape
+  - scale
+  - orientation
+  - position_x
+  - position_y
+  ignored_factors:
+  - foreground_color
+  - background_color
+training:
+  epochs: 1
+  batch_size: 64
+  num_workers: 8
+  num_values: 2
+  num_filters: 64
+  lr: 0.0001
+  seed: 2389822288
+  base_seed: 42
+  bbox_loss_inside_weight: 1.0
+  bbox_loss_outside_weight: 1.0

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/checkpoint_summary.json

@@ -0,0 +1,10 @@
+{
+  "checkpoints": {
+    "best_in_domain": {
+      "monitor": "val_in_domain_bpd",
+      "path": "CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt",
+      "score": 0.03411465883255005
+    }
+  },
+  "default_checkpoint": "best_in_domain"
+}

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/checkpoints/best_in_domain.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c63afbe6a750eea702efaec33013805b82e6cfe811bec36dea057dcbb84026c3
+size 12959407

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/dataset_loader_config.json

@@ -0,0 +1,212 @@
+{
+  "cache": {
+    "available_shapes": [
+      "polygon_3",
+      "polygon_4",
+      "polygon_5",
+      "star_3_r02",
+      "star_4_r04",
+      "star_5_r04",
+      "star_6_r04",
+      "ellipse_050",
+      "heart",
+      "flower_2_w03",
+      "flower_3_w03",
+      "flower_4_w03",
+      "flower_5_w03",
+      "flower_6_w03",
+      "rectangle_03",
+      "cross_065",
+      "crescent_025",
+      "drop_15",
+      "closed_hollow_polygon_3",
+      "closed_hollow_polygon_4",
+      "closed_hollow_polygon_5",
+      "hollow_polygon_3",
+      "hollow_polygon_4",
+      "hollow_polygon_5",
+      "closed_hollow_ellipse_05"
+    ],
+    "generation_config": {
+      "candidate_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "candidate_fg_colors": [
+        [
+          255,
+          255,
+          255
+        ]
+      ],
+      "image_size": 64,
+      "num_values": 2,
+      "ood": {
+        "bg_colors_from": "unseen",
+        "fg_colors_from": "unseen",
+        "num_bg_colors": 0,
+        "num_fg_colors": 0,
+        "num_shapes": 0,
+        "shapes_from": "unseen"
+      },
+      "orientations": {
+        "max": 6.28318531,
+        "min": 0.0,
+        "num": 40
+      },
+      "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+      "output_channels": 1,
+      "position_space": "renderer",
+      "positions_x": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "positions_y": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "scale_space": "renderer",
+      "scales": [
+        0.2,
+        0.24,
+        0.28,
+        0.32,
+        0.36,
+        0.4
+      ],
+      "seed": 42,
+      "shape_pool": "all",
+      "train": {
+        "num_bg_colors": 1,
+        "num_fg_colors": 1,
+        "num_shapes": 25
+      }
+    },
+    "has_factor_structure": false,
+    "image_size": 64,
+    "num_images": 6144000,
+    "num_values": 2,
+    "output_channels": 1,
+    "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+    "split_config": {
+      "bg_colors_from": "unseen",
+      "fg_colors_from": "unseen",
+      "ood_bg_colors": [],
+      "ood_fg_colors": [],
+      "ood_index_range": {
+        "start": 6144000,
+        "stop": 6144000
+      },
+      "ood_shapes": [],
+      "ood_size": 0,
+      "seed": 42,
+      "shapes_from": "unseen",
+      "train_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "train_fg_colors": [
+        [
+          1,
+          1,
+          1
+        ]
+      ],
+      "train_index_range": {
+        "start": 0,
+        "stop": 6144000
+      },
+      "train_shapes": [
+        "polygon_3",
+        "polygon_4",
+        "polygon_5",
+        "star_3_r02",
+        "star_4_r04",
+        "star_5_r04",
+        "star_6_r04",
+        "ellipse_050",
+        "heart",
+        "flower_2_w03",
+        "flower_3_w03",
+        "flower_4_w03",
+        "flower_5_w03",
+        "flower_6_w03",
+        "rectangle_03",
+        "cross_065",
+        "crescent_025",
+        "drop_15",
+        "closed_hollow_polygon_3",
+        "closed_hollow_polygon_4",
+        "closed_hollow_polygon_5",
+        "hollow_polygon_3",
+        "hollow_polygon_4",
+        "hollow_polygon_5",
+        "closed_hollow_ellipse_05"
+      ],
+      "train_size": 6144000
+    }
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "held_out_ood_size": 0,
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ],
+    "in_domain_pool_size": 737280,
+    "ood_excluded_from_pool": true,
+    "ood_excluded_from_training_pool": true,
+    "ood_validation_pool_size": 737280,
+    "ood_validation_uses_combined_pool": false,
+    "split_source": "cache_split",
+    "task_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ]
+  },
+  "dataset_name": "sprites_large_composition",
+  "loader_filter": {
+    "seed": 1247230849,
+    "sprite_restricted_shapes": [
+      "closed_hollow_polygon_4"
+    ],
+    "sprite_right_side_threshold": 0.5,
+    "sprite_shapes": [
+      "closed_hollow_polygon_4",
+      "flower_3_w03",
+      "hollow_polygon_5"
+    ],
+    "val_fraction": 0.0
+  },
+  "splits": {
+    "held_out_ood_size": 0,
+    "in_domain_pool_size": 737280,
+    "ood_validation_pool_size": 737280,
+    "train_anchor_size": 737280,
+    "train_rule": "Train composition on in-domain anchors only; command and target images are sampled from the in-domain pool, and held-out OOD indices are excluded from all roles.",
+    "val_anchor_size": 737280,
+    "val_in_domain_anchor_size": 737280,
+    "val_in_domain_rule": "Validate composition on fixed in-domain anchors only; command and target images are sampled from the same in-domain pool, with OOD indices excluded from all roles.",
+    "val_ood_anchor_size": 0,
+    "val_ood_rule": null,
+    "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."
+  }
+}

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/events.out.tfevents.1775555340.node04.1239199.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ed31420540662634ffaa0cefe306daa50ff28d67f8824f68d3ad5a2fd429f95
+size 46511

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12439/hparams.yaml

@@ -0,0 +1,10 @@
+num_filters: 64
+z_dim: 10
+lr: 0.0001
+composition_op: fixint
+n_actions: 5
+num_channels: 1
+num_values: 2
+image_size: 64
+bbox_loss_inside_weight: 1.0
+bbox_loss_outside_weight: 1.0

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/sweep_run_config.json

@@ -0,0 +1,143 @@
+{
+  "cache": {
+    "available_shapes": [
+      "polygon_3",
+      "polygon_4",
+      "polygon_5",
+      "star_3_r02",
+      "star_4_r04",
+      "star_5_r04",
+      "star_6_r04",
+      "ellipse_050",
+      "heart",
+      "flower_2_w03",
+      "flower_3_w03",
+      "flower_4_w03",
+      "flower_5_w03",
+      "flower_6_w03",
+      "rectangle_03",
+      "cross_065",
+      "crescent_025",
+      "drop_15",
+      "closed_hollow_polygon_3",
+      "closed_hollow_polygon_4",
+      "closed_hollow_polygon_5",
+      "hollow_polygon_3",
+      "hollow_polygon_4",
+      "hollow_polygon_5",
+      "closed_hollow_ellipse_05"
+    ],
+    "generation_config": {
+      "candidate_bg_colors": [
+        [
+          0,
+          0,
+          0
+        ]
+      ],
+      "candidate_fg_colors": [
+        [
+          255,
+          255,
+          255
+        ]
+      ],
+      "image_size": 64,
+      "num_values": 2,
+      "ood": {
+        "bg_colors_from": "unseen",
+        "fg_colors_from": "unseen",
+        "num_bg_colors": 0,
+        "num_fg_colors": 0,
+        "num_shapes": 0,
+        "shapes_from": "unseen"
+      },
+      "orientations": {
+        "max": 6.28318531,
+        "min": 0.0,
+        "num": 40
+      },
+      "output": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt",
+      "output_channels": 1,
+      "position_space": "renderer",
+      "positions_x": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "positions_y": {
+        "max": 0.8,
+        "min": 0.2,
+        "num": 32
+      },
+      "scale_space": "renderer",
+      "scales": [
+        0.2,
+        0.24,
+        0.28,
+        0.32,
+        0.36,
+        0.4
+      ],
+      "seed": 42,
+      "shape_pool": "all",
+      "train": {
+        "num_bg_colors": 1,
+        "num_fg_colors": 1,
+        "num_shapes": 25
+      }
+    },
+    "path": "data/dsprites_large_binary/dsprites_large_6_40_32_32.pt"
+  },
+  "composition": {
+    "action_factors": [
+      "shape",
+      "scale",
+      "orientation",
+      "position_x",
+      "position_y"
+    ],
+    "command_sampler": "lils",
+    "composition_op": "fixint",
+    "ignored_factors": [
+      "foreground_color",
+      "background_color"
+    ]
+  },
+  "dataloader": {
+    "sprite_restricted_shapes": [
+      "closed_hollow_polygon_4"
+    ],
+    "sprite_right_side_threshold": 0.5,
+    "sprite_shapes": [
+      "closed_hollow_polygon_4",
+      "flower_3_w03",
+      "hollow_polygon_5"
+    ],
+    "val_fraction": 0.0
+  },
+  "held_out_shape": "closed_hollow_polygon_4",
+  "n_shapes": 3,
+  "rerun_index": 2,
+  "run_index": 2,
+  "run_name": "run_002_rerun_02_03shapes_zdim_010",
+  "run_seed": 1247230849,
+  "run_shapes": [
+    "closed_hollow_polygon_4",
+    "flower_3_w03",
+    "hollow_polygon_5"
+  ],
+  "training": {
+    "base_seed": 42,
+    "batch_size": 64,
+    "bbox_loss_inside_weight": 1.0,
+    "bbox_loss_outside_weight": 1.0,
+    "epochs": 1,
+    "lr": 0.0001,
+    "num_filters": 64,
+    "num_values": 2,
+    "num_workers": 8,
+    "seed": 1247230849
+  },
+  "z_dim": 10
+}

CompVAE_logs/dsprites_original_large/run_002_rerun_02_03shapes_zdim_010/sweep_run_config.yaml

@@ -0,0 +1,118 @@
+run_index: 2
+rerun_index: 2
+run_name: run_002_rerun_02_03shapes_zdim_010
+run_seed: 1247230849
+z_dim: 10
+n_shapes: 3
+held_out_shape: closed_hollow_polygon_4
+run_shapes:
+- closed_hollow_polygon_4
+- flower_3_w03
+- hollow_polygon_5
+cache:
+  path: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+  available_shapes:
+  - polygon_3
+  - polygon_4
+  - polygon_5
+  - star_3_r02
+  - star_4_r04
+  - star_5_r04
+  - star_6_r04
+  - ellipse_050
+  - heart
+  - flower_2_w03
+  - flower_3_w03
+  - flower_4_w03
+  - flower_5_w03
+  - flower_6_w03
+  - rectangle_03
+  - cross_065
+  - crescent_025
+  - drop_15
+  - closed_hollow_polygon_3
+  - closed_hollow_polygon_4
+  - closed_hollow_polygon_5
+  - hollow_polygon_3
+  - hollow_polygon_4
+  - hollow_polygon_5
+  - closed_hollow_ellipse_05
+  generation_config:
+    seed: 42
+    image_size: 64
+    num_values: 2
+    output_channels: 1
+    output: data/dsprites_large_binary/dsprites_large_6_40_32_32.pt
+    scale_space: renderer
+    position_space: renderer
+    shape_pool: all
+    candidate_fg_colors:
+    - - 255
+      - 255
+      - 255
+    candidate_bg_colors:
+    - - 0
+      - 0
+      - 0
+    scales:
+    - 0.2
+    - 0.24
+    - 0.28
+    - 0.32
+    - 0.36
+    - 0.4
+    orientations:
+      min: 0.0
+      max: 6.28318531
+      num: 40
+    positions_x:
+      min: 0.2
+      max: 0.8
+      num: 32
+    positions_y:
+      min: 0.2
+      max: 0.8
+      num: 32
+    train:
+      num_shapes: 25
+      num_fg_colors: 1
+      num_bg_colors: 1
+    ood:
+      num_shapes: 0
+      shapes_from: unseen
+      num_fg_colors: 0
+      fg_colors_from: unseen
+      num_bg_colors: 0
+      bg_colors_from: unseen
+dataloader:
+  sprite_shapes:
+  - closed_hollow_polygon_4
+  - flower_3_w03
+  - hollow_polygon_5
+  sprite_restricted_shapes:
+  - closed_hollow_polygon_4
+  sprite_right_side_threshold: 0.5
+  val_fraction: 0.0
+composition:
+  composition_op: fixint
+  command_sampler: lils
+  action_factors:
+  - shape
+  - scale
+  - orientation
+  - position_x
+  - position_y
+  ignored_factors:
+  - foreground_color
+  - background_color
+training:
+  epochs: 1
+  batch_size: 64
+  num_workers: 8
+  num_values: 2
+  num_filters: 64
+  lr: 0.0001
+  seed: 1247230849
+  base_seed: 42
+  bbox_loss_inside_weight: 1.0
+  bbox_loss_outside_weight: 1.0

CompVAE_logs/test_sweep/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12272/events.out.tfevents.1775306940.node04.1313920.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b1f92d10eb727f06051ca7108a8ee19e49873036ac7fba13a321e86e5593c36
+size 5009

CompVAE_logs/test_sweep/run_000_rerun_00_03shapes_zdim_010/lightning_logs/version_12331/events.out.tfevents.1775372961.node04.3225156.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:065e80583de4b6d75b80ec7d7ee901fa1b18ed64507d64922348655d79061edb
+size 97545

CompVAE_logs/test_sweep/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12273/events.out.tfevents.1775306948.node05.711225.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67288dd43aa1533b6d54782d6add337f6d902206c8e1d233044bd030ff67c00b
+size 5088

CompVAE_logs/test_sweep/run_000_rerun_00_06shapes_zdim_010/lightning_logs/version_12332/events.out.tfevents.1775372974.node04.3226990.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:342a8718ad16626d55c8b784a20419159b4b02342a61b6c836e1fa75b3e9c5cd
+size 95134

CompVAE_logs/test_sweep/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12274/events.out.tfevents.1775306923.node05.711387.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c8d0574f3407b8a08ea4f149a1d4dcc0a315b77639b9483e44c8a964b80e36f
+size 5159

CompVAE_logs/test_sweep/run_000_rerun_00_09shapes_zdim_010/lightning_logs/version_12333/events.out.tfevents.1775372909.node05.885931.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87b11d9c3649e0449c1bbf2eb1eba90ed9ebe96965753e8dcd23351cfd8187ae
+size 95454

CompVAE_logs/test_sweep/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12275/events.out.tfevents.1775306938.node05.711531.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e9c6f42c40bda0ee7b36fd58a5cae4ec1e39e5f4f866e72457c9fe375565d0c
+size 5235

CompVAE_logs/test_sweep/run_000_rerun_00_12shapes_zdim_010/lightning_logs/version_12334/events.out.tfevents.1775372906.node05.886182.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5424f48ed2967e22820dc7e2c70afb04f6ed3504326f581d1a3dbb90ebc0e763
+size 102527

CompVAE_logs/test_sweep/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12276/events.out.tfevents.1775306982.node06.1570303.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a2667339e442b02fed2e33d570378111ce447a86fe49362890d85d2bc18f6584
+size 5311

CompVAE_logs/test_sweep/run_000_rerun_00_15shapes_zdim_010/lightning_logs/version_12335/events.out.tfevents.1775372906.node05.886329.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:127fc47d118c4039f498a682e22a739a0789a1af4af801d46cbc36e0902561e4
+size 102122

CompVAE_logs/test_sweep/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12277/events.out.tfevents.1775306974.node06.1570455.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a4c80703567dbf90bcd3db3386c61bee94d659c6bd30508de1ed995ee791df1
+size 5412

CompVAE_logs/test_sweep/run_000_rerun_00_20shapes_zdim_010/lightning_logs/version_12336/events.out.tfevents.1775372924.node05.886485.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dbce119a419fcd66192d121c219edaa66b712635183506a298cf977c1d101270
+size 99973

CompVAE_logs/test_sweep/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12278/events.out.tfevents.1775306968.node06.1570614.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:322d4ce87e7335c203b7200f7e03bf32dcfd8a11dd4afb34eefb06ad6517fef6
+size 5534

CompVAE_logs/test_sweep/run_000_rerun_00_25shapes_zdim_010/lightning_logs/version_12337/events.out.tfevents.1775372917.node05.886628.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b71d4ef4acda7344c3248b62a7d1eb4a83278c96d39d21c226002f15f8d5e6a5
+size 101048

CompVAE_logs/test_sweep/run_001_rerun_01_03shapes_zdim_010/lightning_logs/version_12331/events.out.tfevents.1775377811.node04.3225156.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83be2bfc642e10a23d3ce0921d28aa6e2a934cce0f04171bec54deb42f68750d
+size 101276

CompVAE_logs/test_sweep/run_001_rerun_01_06shapes_zdim_010/lightning_logs/version_12332/events.out.tfevents.1775382149.node04.3226990.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2098a06a136c213b33bc6b98d12e2c99d8faca60e644eac4ea5d07414db9e764
+size 99545

CompVAE_logs/test_sweep/run_001_rerun_01_09shapes_zdim_010/lightning_logs/version_12333/events.out.tfevents.1775386399.node05.885931.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:994a720e7ac9a6f70d9d91550bc6140443b34261bc37137a9cb650010e682e20
+size 101498

CompVAE_logs/test_sweep/run_001_rerun_01_12shapes_zdim_010/lightning_logs/version_12334/events.out.tfevents.1775390703.node05.886182.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37aa46683877b02dfad062aae74dc8dd97811674d86a6d39dd7f9f09c7d75c37
+size 97591

CompVAE_logs/test_sweep/run_001_rerun_01_15shapes_zdim_010/lightning_logs/version_12335/events.out.tfevents.1775395032.node05.886329.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e3b5f9738959b60a0574888545ab7201d565b6099d440ff59a601ec390020f55
+size 101771

CompVAE_logs/test_sweep/run_001_rerun_01_20shapes_zdim_010/lightning_logs/version_12336/events.out.tfevents.1775402320.node05.886485.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:122049879e7a04fa8e9fe98321440eebe5170bf49e03d772d08378bc1580431f
+size 100371

CompVAE_logs/test_sweep/run_001_rerun_01_25shapes_zdim_010/lightning_logs/version_12337/events.out.tfevents.1775409681.node05.886628.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:065308ffec5acf92f7492c598b3a38d4682be67947ca59fbb5c74bf94e11db5d
+size 96857

CompVAE_logs/test_sweep/run_002_rerun_02_03shapes_zdim_010/lightning_logs/version_12331/events.out.tfevents.1775382673.node04.3225156.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5fe635060ea1ec1a7e3f4a9b7b4ee5a85b185a7e1e466809ad9ff3e27edc1543
+size 110523

CompVAE_logs/test_sweep/run_002_rerun_02_06shapes_zdim_010/lightning_logs/version_12332/events.out.tfevents.1775391298.node04.3226990.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f5132ad655fe8f89040de0b6fd88f06d56aa8fb3a2297568fbd4decf3dffbf05
+size 101561

CompVAE_logs/test_sweep/run_002_rerun_02_09shapes_zdim_010/lightning_logs/version_12333/events.out.tfevents.1775399863.node05.885931.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d94f7214e259fd9e8cda261558c74c453f9ceacea9b84a0ac40b4d063f55c7a5
+size 103848

CompVAE_logs/test_sweep/run_002_rerun_02_12shapes_zdim_010/lightning_logs/version_12334/events.out.tfevents.1775408514.node05.886182.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e598921d3c60b659c5c8823b17b26062e50232d185f67efe7613a386e9282de4
+size 106391

CompVAE_logs/test_sweep/run_002_rerun_02_15shapes_zdim_010/lightning_logs/version_12335/events.out.tfevents.1775417155.node05.886329.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b779561ee04027e4bca60a5748136535d536fd7e586b524ed6c7c72be824897e
+size 98330