| { |
| "_class_name": "Model", |
| "_diffusers_version": "0.32.2", |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.Model", |
| "full_cfg": { |
| "callbacks": { |
| "vis": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.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": 2, |
| "num_workers": 4, |
| "test_batch_size": 2, |
| "testset_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.ModelDset", |
| "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/.m0227_aekl_rgba_v1_test/", |
| "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/", |
| "downsample": 2, |
| "duplicate": 1, |
| "max_n_patch": 32, |
| "max_seq_len": 4096, |
| "metadf_path": "/home/xya120/studio/datasets/ABO/somages/df_p512_m002_.8sqrtrescale_meta.json", |
| "mode": "allcateval10|ace10transpose", |
| "no_cache": true |
| }, |
| "trainset_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.ModelDset", |
| "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/.m0227_aekl_rgba_v1/", |
| "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/", |
| "downsample": 2, |
| "duplicate": 1, |
| "metadf_path": "/home/xya120/studio/datasets/ABO/somages/df_p512_m002_.8sqrtrescale_meta.json", |
| "mode": "", |
| "no_cache": true |
| } |
| }, |
| "expr_name": "somages/m0227_aekl_rgba_v1", |
| "meta_info": { |
| "checkpoints_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/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/m0227_aekl_rgba_v1", |
| "logs_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/logs", |
| "results_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/results", |
| "session_name": "somages/m0227_aekl_rgba_v1_250313_2233", |
| "src_dir": "/home/xya120/studio/omages_internal/src" |
| }, |
| "pl_model_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.Model", |
| "full_opt": { |
| "callbacks": { |
| "vis": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.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": 2, |
| "num_workers": 4, |
| "test_batch_size": 2, |
| "testset_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.ModelDset", |
| "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/.m0227_aekl_rgba_v1_test/", |
| "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/", |
| "downsample": 2, |
| "duplicate": 1, |
| "max_n_patch": 32, |
| "max_seq_len": 4096, |
| "metadf_path": "/home/xya120/studio/datasets/ABO/somages/df_p512_m002_.8sqrtrescale_meta.json", |
| "mode": "allcateval10|ace10transpose", |
| "no_cache": true |
| }, |
| "trainset_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.ModelDset", |
| "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/.m0227_aekl_rgba_v1/", |
| "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/", |
| "downsample": 2, |
| "duplicate": 1, |
| "metadf_path": "/home/xya120/studio/datasets/ABO/somages/df_p512_m002_.8sqrtrescale_meta.json", |
| "mode": "", |
| "no_cache": true |
| } |
| }, |
| "expr_name": "somages/m0227_aekl_rgba_v1", |
| "meta_info": { |
| "checkpoints_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/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/m0227_aekl_rgba_v1", |
| "logs_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/logs", |
| "results_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/results", |
| "session_name": "somages/m0227_aekl_rgba_v1_250313_2233", |
| "src_dir": "/home/xya120/studio/omages_internal/src" |
| }, |
| "pl_model_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.Model", |
| "model": { |
| "network": { |
| "channel_input": 3, |
| "channel_output": 3, |
| "debug_boundary": false, |
| "model_id": "black-forest-labs/FLUX.1-schnell", |
| "train_encoder": true |
| }, |
| "optimizer": { |
| "_target_": "torch.optim.AdamW", |
| "lr": 1e-05 |
| }, |
| "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" |
| }, |
| "model": { |
| "network": { |
| "channel_input": 3, |
| "channel_output": 3, |
| "debug_boundary": false, |
| "model_id": "black-forest-labs/FLUX.1-schnell", |
| "train_encoder": true |
| }, |
| "optimizer": { |
| "_target_": "torch.optim.AdamW", |
| "lr": 1e-05 |
| }, |
| "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.m0227_aekl_rgba_v1.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": 2, |
| "num_workers": 4, |
| "test_batch_size": 2, |
| "testset_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.ModelDset", |
| "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/.m0227_aekl_rgba_v1_test/", |
| "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/", |
| "downsample": 2, |
| "duplicate": 1, |
| "max_n_patch": 32, |
| "max_seq_len": 4096, |
| "metadf_path": "/home/xya120/studio/datasets/ABO/somages/df_p512_m002_.8sqrtrescale_meta.json", |
| "mode": "allcateval10|ace10transpose", |
| "no_cache": true |
| }, |
| "trainset_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.ModelDset", |
| "cache_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/.m0227_aekl_rgba_v1/", |
| "data_root": "/home/xya120/studio/datasets/ABO/somages/v3v1/", |
| "downsample": 2, |
| "duplicate": 1, |
| "metadf_path": "/home/xya120/studio/datasets/ABO/somages/df_p512_m002_.8sqrtrescale_meta.json", |
| "mode": "", |
| "no_cache": true |
| } |
| }, |
| "expr_name": "somages/m0227_aekl_rgba_v1", |
| "meta_info": { |
| "checkpoints_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/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/m0227_aekl_rgba_v1", |
| "logs_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/logs", |
| "results_dir": "/home/xya120/studio/omages_internal/experiments/somages/m0227_aekl_rgba_v1/results", |
| "session_name": "somages/m0227_aekl_rgba_v1_250313_2233", |
| "src_dir": "/home/xya120/studio/omages_internal/src" |
| }, |
| "pl_model_opt": { |
| "_target_": "src.models.omages_flux.m0227_aekl_rgba_v1.Model", |
| "model": { |
| "network": { |
| "channel_input": 3, |
| "channel_output": 3, |
| "debug_boundary": false, |
| "model_id": "black-forest-labs/FLUX.1-schnell", |
| "train_encoder": true |
| }, |
| "optimizer": { |
| "_target_": "torch.optim.AdamW", |
| "lr": 1e-05 |
| }, |
| "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": true, |
| "model": { |
| "network": { |
| "channel_input": 3, |
| "channel_output": 3, |
| "debug_boundary": false, |
| "model_id": "black-forest-labs/FLUX.1-schnell", |
| "train_encoder": true |
| }, |
| "optimizer": { |
| "_target_": "torch.optim.AdamW", |
| "lr": 1e-05 |
| }, |
| "scheduler": { |
| "T_max": 600, |
| "_target_": "torch.optim.lr_scheduler.CosineAnnealingLR", |
| "eta_min": 1e-07 |
| } |
| }, |
| "scaling_factor": 0.01 |
| } |
|
|
