| { | |
| "_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_2220", | |
| "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_2220", | |
| "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_2220", | |
| "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 | |
| } | |
| }, | |
| "networks": {}, | |
| "optimizer": {}, | |
| "scaling_factor": 0.01, | |
| "scheduler": {} | |
| } | |