Skip to content

Generator Model Training Orchestration#27

Merged
MattsonCam merged 18 commits into
generator_modelsfrom
review_generator_models
Jul 13, 2026
Merged

Generator Model Training Orchestration#27
MattsonCam merged 18 commits into
generator_modelsfrom
review_generator_models

Conversation

@MattsonCam

@MattsonCam MattsonCam commented Jul 1, 2026

Copy link
Copy Markdown
Member

This PR expands the generator-model training workflow into a more reusable orchestration pattern and simplifies the training stack around the current reconstruction-based approach. It adds broader run configuration for optimization, checkpointing, resume behavior, and training/evaluation precision control, while also improving deterministic data handling and evaluation behavior. The result is a cleaner foundation for training current and future generator models with more consistent experiment management.

@MattsonCam MattsonCam requested a review from wli51 July 1, 2026 15:47
@review-notebook-app

Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Cameron Mattson added 12 commits July 1, 2026 11:35
Experiment ID:2
Run ID:5e9a052521da4320871db9625e09cf78
Run Name:agreeable-turtle-117

- ConvNeXtUNet Generator
- Single 2D crop input and single 2D crop target
- Cache-backed filtered nucleus crops generated from the configured data directory
- L1 optimization objective with L2, PSNR, SSIM, and Pearson correlation metric logging
- Normalization of input and target images using the max possible pixel intensity
- Training on entire initial dataset
- Sigmoid output activation function when computing metrics and loss and saving images

@wli51 wli51 left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@MattsonCam great work! This is a very difficult dataset to train on. Hopefully improvements but I wouldn't be surprised if the dynamic range still breaks everything.

Comment thread train.py Outdated
Comment thread README.md Outdated
Comment thread README.md
@MattsonCam

Copy link
Copy Markdown
Member Author

Thanks for the review @wli51 ! There will be a couple more prs shortly. Merging now

@MattsonCam MattsonCam merged commit a54a08d into generator_models Jul 13, 2026
@MattsonCam MattsonCam deleted the review_generator_models branch July 13, 2026 20:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants