{
  "_class_name": "Model",
  "_diffusers_version": "0.32.2",
  "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Model",
  "full_cfg": {
    "callbacks": {
      "vis": {
        "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Visualizer",
        "every_n_epoch": 2,
        "infer_steps": "250",
        "load_compute": false,
        "mode": "N2G",
        "no_sanity_check": false,
        "parallel_vis": true,
        "visual_indices": [
          0,
          1,
          2,
          3
        ]
      }
    },
    "datamodule_opt": {
      "_target_": "src.datamodule.DataModule",
      "batch_size": 4,
      "num_workers": 4,
      "test_batch_size": 4,
      "testset_opt": {
        "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.ModelDset",
        "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/.m0425_aekl_v2_xyzob_test/",
        "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/",
        "df_subsamples": null,
        "downsample": 2,
        "duplicate": 1,
        "max_n_patch": 32,
        "max_seq_len": 4096,
        "metadf_path": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/df_v3v2_008m128p_final.json",
        "mode": "allcateval10|ace10transpose|4chan",
        "no_cache": true
      },
      "trainset_opt": {
        "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.ModelDset",
        "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/.m0425_aekl_v2_xyzob_train/",
        "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/",
        "df_subsamples": null,
        "downsample": 2,
        "duplicate": 1,
        "metadf_path": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/df_v3v2_008m128p_final.json",
        "mode": "4chan|random_transform",
        "no_cache": true
      }
    },
    "expr_name": "somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf",
    "meta_info": {
      "checkpoints_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/checkpoints",
      "datasets_dir": "/home/xya120/studio/omages_internal/datasets/",
      "experiments_dir": "/home/xya120/studio/omages_internal/experiments/",
      "expr_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf",
      "logs_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/logs",
      "results_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/results",
      "session_name": "somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf_250707_0911",
      "src_dir": "/home/xya120/studio/omages_internal/src"
    },
    "pl_model_opt": {
      "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Model",
      "full_opt": {
        "callbacks": {
          "vis": {
            "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Visualizer",
            "every_n_epoch": 2,
            "infer_steps": "250",
            "load_compute": false,
            "mode": "N2G",
            "no_sanity_check": false,
            "parallel_vis": true,
            "visual_indices": [
              0,
              1,
              2,
              3
            ]
          }
        },
        "datamodule_opt": {
          "_target_": "src.datamodule.DataModule",
          "batch_size": 4,
          "num_workers": 4,
          "test_batch_size": 4,
          "testset_opt": {
            "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.ModelDset",
            "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/.m0425_aekl_v2_xyzob_test/",
            "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/",
            "df_subsamples": null,
            "downsample": 2,
            "duplicate": 1,
            "max_n_patch": 32,
            "max_seq_len": 4096,
            "metadf_path": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/df_v3v2_008m128p_final.json",
            "mode": "allcateval10|ace10transpose|4chan",
            "no_cache": true
          },
          "trainset_opt": {
            "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.ModelDset",
            "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/.m0425_aekl_v2_xyzob_train/",
            "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/",
            "df_subsamples": null,
            "downsample": 2,
            "duplicate": 1,
            "metadf_path": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/df_v3v2_008m128p_final.json",
            "mode": "4chan|random_transform",
            "no_cache": true
          }
        },
        "expr_name": "somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf",
        "meta_info": {
          "checkpoints_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/checkpoints",
          "datasets_dir": "/home/xya120/studio/omages_internal/datasets/",
          "experiments_dir": "/home/xya120/studio/omages_internal/experiments/",
          "expr_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf",
          "logs_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/logs",
          "results_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/results",
          "session_name": "somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf_250707_0911",
          "src_dir": "/home/xya120/studio/omages_internal/src"
        },
        "pl_model_opt": {
          "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Model",
          "mode": "mse|normalloss",
          "model": {
            "load_model_subfolder": false,
            "network": {
              "channel_input": 4,
              "channel_output": 4,
              "debug_boundary": false,
              "kl_loss_scale": 0.0,
              "model_id": "3dlg-hcvc/m0425_aekl_v2_xyzo_mse",
              "train_encoder": true
            },
            "optimizer": {
              "_target_": "torch.optim.AdamW",
              "lr": 1e-05
            },
            "scaling_factor": 0.1,
            "scheduler": {
              "T_max": 600,
              "_target_": "torch.optim.lr_scheduler.CosineAnnealingLR",
              "eta_min": 1e-07
            }
          }
        },
        "pltrainer_opt": {
          "accelerator": "auto",
          "auto_lr_find": false,
          "check_val_every_n_epoch": 2,
          "checkpoint_monitor": "val/loss",
          "copy_ckpt_from": "",
          "disable_auto_lr_scale": false,
          "early_stop_patience": 1000,
          "extra_kwargs": {
            "enable_model_summary": true,
            "num_sanity_val_steps": 0,
            "precision": "16-mixed"
          },
          "fast_dev_run": false,
          "gpus": [
            0
          ],
          "gradient_clip_val": null,
          "logger": "wandb",
          "logger_kwargs": {
            "mode": "online",
            "notes": "aekl_geometry",
            "tags": [
              "aekl_geometry"
            ]
          },
          "max_epochs": 600,
          "max_time": null,
          "min_epochs": 1,
          "precision": "32-true",
          "resume_from": "resume",
          "seed": 314,
          "strategy": "ddp_find_unused_parameters_true",
          "strict_loading": true,
          "use_tensorcores": true,
          "wandb_silent": false
        },
        "project_name": "omg"
      },
      "mode": "mse|normalloss",
      "model": {
        "load_model_subfolder": false,
        "network": {
          "channel_input": 4,
          "channel_output": 4,
          "debug_boundary": false,
          "kl_loss_scale": 0.0,
          "model_id": "3dlg-hcvc/m0425_aekl_v2_xyzo_mse",
          "train_encoder": true
        },
        "optimizer": {
          "_target_": "torch.optim.AdamW",
          "lr": 1e-05
        },
        "scaling_factor": 0.1,
        "scheduler": {
          "T_max": 600,
          "_target_": "torch.optim.lr_scheduler.CosineAnnealingLR",
          "eta_min": 1e-07
        }
      }
    },
    "pltrainer_opt": {
      "accelerator": "auto",
      "auto_lr_find": false,
      "check_val_every_n_epoch": 2,
      "checkpoint_monitor": "val/loss",
      "copy_ckpt_from": "",
      "disable_auto_lr_scale": false,
      "early_stop_patience": 1000,
      "extra_kwargs": {
        "enable_model_summary": true,
        "num_sanity_val_steps": 0,
        "precision": "16-mixed"
      },
      "fast_dev_run": false,
      "gpus": [
        0
      ],
      "gradient_clip_val": null,
      "logger": "wandb",
      "logger_kwargs": {
        "mode": "online",
        "notes": "aekl_geometry",
        "tags": [
          "aekl_geometry"
        ]
      },
      "max_epochs": 600,
      "max_time": null,
      "min_epochs": 1,
      "precision": "32-true",
      "resume_from": "resume",
      "seed": 314,
      "strategy": "ddp_find_unused_parameters_true",
      "strict_loading": true,
      "use_tensorcores": true,
      "wandb_silent": false
    },
    "project_name": "omg"
  },
  "full_opt": {
    "callbacks": {
      "vis": {
        "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Visualizer",
        "every_n_epoch": 2,
        "infer_steps": "250",
        "load_compute": false,
        "mode": "N2G",
        "no_sanity_check": false,
        "parallel_vis": true,
        "visual_indices": [
          0,
          1,
          2,
          3
        ]
      }
    },
    "datamodule_opt": {
      "_target_": "src.datamodule.DataModule",
      "batch_size": 4,
      "num_workers": 4,
      "test_batch_size": 4,
      "testset_opt": {
        "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.ModelDset",
        "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/.m0425_aekl_v2_xyzob_test/",
        "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/",
        "df_subsamples": null,
        "downsample": 2,
        "duplicate": 1,
        "max_n_patch": 32,
        "max_seq_len": 4096,
        "metadf_path": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/df_v3v2_008m128p_final.json",
        "mode": "allcateval10|ace10transpose|4chan",
        "no_cache": true
      },
      "trainset_opt": {
        "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.ModelDset",
        "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/.m0425_aekl_v2_xyzob_train/",
        "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/",
        "df_subsamples": null,
        "downsample": 2,
        "duplicate": 1,
        "metadf_path": "/home/xya120/studio/datasets/ABO/somages/v3v2_008m128p/df_v3v2_008m128p_final.json",
        "mode": "4chan|random_transform",
        "no_cache": true
      }
    },
    "expr_name": "somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf",
    "meta_info": {
      "checkpoints_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/checkpoints",
      "datasets_dir": "/home/xya120/studio/omages_internal/datasets/",
      "experiments_dir": "/home/xya120/studio/omages_internal/experiments/",
      "expr_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf",
      "logs_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/logs",
      "results_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf/results",
      "session_name": "somages/m0425_aekl_v2_xyzob_mse_normalloss_randtransf_250707_0911",
      "src_dir": "/home/xya120/studio/omages_internal/src"
    },
    "pl_model_opt": {
      "_target_": "src.models.omages_flux.m0425_aekl_v2_xyzob.Model",
      "mode": "mse|normalloss",
      "model": {
        "load_model_subfolder": false,
        "network": {
          "channel_input": 4,
          "channel_output": 4,
          "debug_boundary": false,
          "kl_loss_scale": 0.0,
          "model_id": "3dlg-hcvc/m0425_aekl_v2_xyzo_mse",
          "train_encoder": true
        },
        "optimizer": {
          "_target_": "torch.optim.AdamW",
          "lr": 1e-05
        },
        "scaling_factor": 0.1,
        "scheduler": {
          "T_max": 600,
          "_target_": "torch.optim.lr_scheduler.CosineAnnealingLR",
          "eta_min": 1e-07
        }
      }
    },
    "pltrainer_opt": {
      "accelerator": "auto",
      "auto_lr_find": false,
      "check_val_every_n_epoch": 2,
      "checkpoint_monitor": "val/loss",
      "copy_ckpt_from": "",
      "disable_auto_lr_scale": false,
      "early_stop_patience": 1000,
      "extra_kwargs": {
        "enable_model_summary": true,
        "num_sanity_val_steps": 0,
        "precision": "16-mixed"
      },
      "fast_dev_run": false,
      "gpus": [
        0
      ],
      "gradient_clip_val": null,
      "logger": "wandb",
      "logger_kwargs": {
        "mode": "online",
        "notes": "aekl_geometry",
        "tags": [
          "aekl_geometry"
        ]
      },
      "max_epochs": 600,
      "max_time": null,
      "min_epochs": 1,
      "precision": "32-true",
      "resume_from": "resume",
      "seed": 314,
      "strategy": "ddp_find_unused_parameters_true",
      "strict_loading": true,
      "use_tensorcores": true,
      "wandb_silent": false
    },
    "project_name": "omg"
  },
  "load_model_subfolder": false,
  "mode": "mse|normalloss",
  "model": {
    "load_model_subfolder": false,
    "network": {
      "channel_input": 4,
      "channel_output": 4,
      "debug_boundary": false,
      "kl_loss_scale": 0.0,
      "model_id": "3dlg-hcvc/m0425_aekl_v2_xyzo_mse",
      "train_encoder": true
    },
    "optimizer": {
      "_target_": "torch.optim.AdamW",
      "lr": 1e-05
    },
    "scaling_factor": 0.1,
    "scheduler": {
      "T_max": 600,
      "_target_": "torch.optim.lr_scheduler.CosineAnnealingLR",
      "eta_min": 1e-07
    }
  },
  "scaling_factor": 0.1
}