Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Search
/
Sign in
Sign up
Appearance settings
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
WayScience
/
virtual_stain_flow
Public
Notifications
You must be signed in to change notification settings
Fork
1
Star
2
Code
Issues
7
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Security and quality
Insights
Add auto logging of trainer metadata and optimizer state saving to help with resumation of training
- #39
#39
Merged
wli51
merged 10 commits into
WayScience:main
WayScience/virtual_stain_flow:main
from
wli51:resume-training
wli51/virtual_stain_flow:resume-training
Copy head branch name to clipboard
Sep 11, 2026
Conversation
Commits
10
(10)
Checks
Files changed
Merged
Add auto logging of trainer metadata and optimizer state saving to help with resumation of training
#39
wli51
merged 10 commits into
WayScience:main
WayScience/virtual_stain_flow:main
from
wli51:resume-training
wli51/virtual_stain_flow:resume-training
Copy head branch name to clipboard
Commits
Commits on Aug 6, 2026
Add optimizer state saving functionality as new trainer contract, update existing properties by best effort inferring some. Remove some confusing and unused properties. Adjust model saving defaults.
wli51
committed
b8000e8
View commit details
Copy full SHA for b8000e8
Browse repository at this point
Implement AutoTrainerLogger for adding auto logging of certain useful trainer metadata and couple logging of model weights by logging of optimizer state to facilitate resuming of complete trainings…
Show description for 3c19922
wli51
committed
3c19922
View commit details
Copy full SHA for 3c19922
Browse repository at this point
Add unit tests for model and optimizer state saving functionality
wli51
committed
bd3036b
View commit details
Copy full SHA for bd3036b
Browse repository at this point
Commits on Aug 7, 2026
Fix MlflowLogger to use _log_trainer_artifact for model weight logging and add unit tests for AbstractTrainer save methods
wli51
committed
7689071
View commit details
Copy full SHA for 7689071
Browse repository at this point
Commits on Sep 9, 2026
Rename _log_trainer_artifact function name with _save_and_log_trainer_artifacts, added more detailed docstring, for improved naming clarity
wli51
committed
3b0d53b
View commit details
Copy full SHA for 3b0d53b
Browse repository at this point
Update DatasetManifest serialization to stop using depracated functions, updating dependency as well.
wli51
committed
f6248b9
View commit details
Copy full SHA for f6248b9
Browse repository at this point
Refactor protocol and abstract class definition to make the `file_name_prefix` parameter of the save_model and save_optimizer_state function non-optional, and directly assign default values in real…
Show description for ac2c0b8
wli51
committed
ac2c0b8
View commit details
Copy full SHA for ac2c0b8
Browse repository at this point
Remove unused _epochs attribute from TrainerProtocol class
wli51
committed
cc3ab85
View commit details
Copy full SHA for cc3ab85
Browse repository at this point
Refactor AbstractTrainer to enforce integer type for epoch parameter and add validation. Introduce utility method for dataset size retrieval and enhance tests for epoch handling and dataset size sc…
Show description for 309927a
wli51
committed
309927a
View commit details
Copy full SHA for 309927a
Browse repository at this point
Commits on Sep 11, 2026
Update CHANGELOG.md for version 0.4.9: Add initial training resume support and fix deprecated Pandas API usage
wli51
committed
6b13039
View commit details
Copy full SHA for 6b13039
Browse repository at this point
You can’t perform that action at this time.
