Skip to content

Save all pytorch config updates in dlc_torch.train_network() - #3062

Merged
AlexEMG merged 2 commits into
mainfrom
maxim/force_save_pytorch_cfg
Jul 24, 2025
Merged

Save all pytorch config updates in dlc_torch.train_network()#3062
AlexEMG merged 2 commits into
mainfrom
maxim/force_save_pytorch_cfg

Conversation

@maximpavliv

@maximpavliv maximpavliv commented Jul 18, 2025

Copy link
Copy Markdown
Contributor

This Pull Request saves all updates done to the model configuration to the disk (pytorch_config.yaml) before starting training in dlc_torch.train_network().

Previously, the pytorch config was only saved when pytorch_cfg_updates were passed. This prevented saving other updated fields.

@AlexEMG

AlexEMG commented Jul 22, 2025

Copy link
Copy Markdown
Member

Thank you! Please update so it only saves if a change was made.

@AlexEMG
AlexEMG self-requested a review July 22, 2025 11:32
@maximpavliv

Copy link
Copy Markdown
Contributor Author

Please update so it only saves if a change was made.

Done!

@maximpavliv maximpavliv changed the title Force save pytorch config in dlc_torch.train_network() Save all pytorch config updates in dlc_torch.train_network() Jul 22, 2025
@AlexEMG
AlexEMG merged commit b400ad9 into main Jul 24, 2025
5 checks passed
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