Added result 16.9248 and 17.3305

This commit is contained in:
2026-01-25 20:39:48 +01:00
parent 9bf3335da6
commit 5545a2f0eb
6 changed files with 4 additions and 4 deletions

Binary file not shown.

Binary file not shown.

View File

@@ -24,17 +24,17 @@ if __name__ == '__main__':
config_dict['results_path'] = os.path.join(project_root, "results")
config_dict['data_path'] = os.path.join(project_root, "data", "dataset")
config_dict['device'] = None
config_dict['learningrate'] = 8e-4 # Higher max LR for OneCycleLR
config_dict['learningrate'] = 1e-3 # Higher max LR for OneCycleLR
config_dict['weight_decay'] = 1e-5 # Lower for faster learning
config_dict['n_updates'] = 3500 # Reduced for fast training
config_dict['batchsize'] = 16 # Larger batch for speed
config_dict['early_stopping_patience'] = 12 # Adjusted patience
config_dict['batchsize'] = 96 # Larger batch for speed
config_dict['early_stopping_patience'] = 3 # Adjusted patience
config_dict['use_wandb'] = False
config_dict['print_train_stats_at'] = 10
config_dict['print_stats_at'] = 100
config_dict['plot_at'] = 500
config_dict['validate_at'] = 150 # More frequent validation
config_dict['validate_at'] = 500 # More frequent validation
network_config = {
'n_in_channels': 4,