tikaiz/DSAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update runtime configuration and model parameters for improved training efficiency

Browse Source
This commit is contained in:
Tim Kainz
2026-01-31 22:35:48 +01:00
parent 4af674b79d
commit d979c200f9
4 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
image-inpainting/src/train.py
View File

@@ -313,8 +313,8 @@ def train(seed, testset_ratio, validset_ratio, data_path, results_path, early_st
input, target = input.to(device), target.to(device)
# Check for runtime config updates every 50 steps
if i % 50 == 0 and i > 0:
# Check for runtime config updates every 5 steps
if i % 5 == 0 and i > 0:
runtime_params = load_runtime_config(config_json_path, runtime_params)
n_updates = runtime_params['n_updates']
plot_at = runtime_params['plot_at']