From 16cdc2ed4e463453fe0c1f14d8a4f2adae5199bc Mon Sep 17 00:00:00 2001 From: Tim Kainz Date: Sat, 24 Jan 2026 16:15:34 +0100 Subject: [PATCH] improve baseline --- image-inpainting/src/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image-inpainting/src/main.py b/image-inpainting/src/main.py index 0c23310..dc4f59c 100644 --- a/image-inpainting/src/main.py +++ b/image-inpainting/src/main.py @@ -25,7 +25,7 @@ if __name__ == '__main__': config_dict['device'] = None config_dict['learningrate'] = 3e-4 # Optimal learning rate for AdamW 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['early_stopping_patience'] = 10 # More patience for complex model config_dict['use_wandb'] = False