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

improve baseline

Browse Source
This commit is contained in:
Tim Kainz
2026-01-24 16:15:34 +01:00
parent 57026695d4
commit 16cdc2ed4e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
image-inpainting/src/main.py
View File

@@ -25,7 +25,7 @@ if __name__ == '__main__':
config_dict['device'] = None config_dict['device'] = None
config_dict['learningrate'] = 3e-4 # Optimal learning rate for AdamW config_dict['learningrate'] = 3e-4 # Optimal learning rate for AdamW
config_dict['weight_decay'] = 1e-4 # Slightly higher for better regularization config_dict['weight_decay'] = 1e-4 # Slightly higher for better regularization
config_dict['n_updates'] = 300 # More updates for better convergence config_dict['n_updates'] = 5000 # More updates for better convergence
config_dict['batchsize'] = 8 # Smaller batch for better gradient estimates config_dict['batchsize'] = 8 # Smaller batch for better gradient estimates
config_dict['early_stopping_patience'] = 10 # More patience for complex model config_dict['early_stopping_patience'] = 10 # More patience for complex model
config_dict['use_wandb'] = False config_dict['use_wandb'] = False