Emily commited on
Commit
cc77288
·
1 Parent(s): f826c3b

Update default epochs to 30 for all datasets and architectures

Browse files

- Increase epochs slider maximum from 20 to 50
- Set default epochs value to 30 in UI
- Update all preset configurations (High Privacy, Balanced, High Utility) to use 30 epochs
- Update setOptimalParameters function to use 30 epochs
- Ensures consistent training duration across all dataset/architecture combinations

app/static/js/main.js CHANGED
@@ -71,21 +71,21 @@ class DPSGDExplorer {
71
  noiseMultiplier: 1.5,
72
  batchSize: 256,
73
  learningRate: 0.005,
74
- epochs: 10
75
  },
76
  'balanced': {
77
  clippingNorm: 1.0,
78
  noiseMultiplier: 1.0,
79
  batchSize: 128,
80
  learningRate: 0.01,
81
- epochs: 8
82
  },
83
  'high-utility': {
84
  clippingNorm: 1.5,
85
  noiseMultiplier: 0.5,
86
  batchSize: 64,
87
  learningRate: 0.02,
88
- epochs: 5
89
  }
90
  };
91
 
@@ -847,7 +847,7 @@ function setOptimalParameters() {
847
  document.getElementById('noise-multiplier').value = '1.0'; // Moderate noise for good privacy
848
  document.getElementById('batch-size').value = '256'; // Large batches for DP-SGD stability
849
  document.getElementById('learning-rate').value = '0.05'; // Balanced learning rate
850
- document.getElementById('epochs').value = '15'; // Sufficient epochs for convergence
851
 
852
  // Update displays
853
  updateClippingNormDisplay();
 
71
  noiseMultiplier: 1.5,
72
  batchSize: 256,
73
  learningRate: 0.005,
74
+ epochs: 30
75
  },
76
  'balanced': {
77
  clippingNorm: 1.0,
78
  noiseMultiplier: 1.0,
79
  batchSize: 128,
80
  learningRate: 0.01,
81
+ epochs: 30
82
  },
83
  'high-utility': {
84
  clippingNorm: 1.5,
85
  noiseMultiplier: 0.5,
86
  batchSize: 64,
87
  learningRate: 0.02,
88
+ epochs: 30
89
  }
90
  };
91
 
 
847
  document.getElementById('noise-multiplier').value = '1.0'; // Moderate noise for good privacy
848
  document.getElementById('batch-size').value = '256'; // Large batches for DP-SGD stability
849
  document.getElementById('learning-rate').value = '0.05'; // Balanced learning rate
850
+ document.getElementById('epochs').value = '30'; // Sufficient epochs for convergence
851
 
852
  // Update displays
853
  updateClippingNormDisplay();
app/templates/index.html CHANGED
@@ -141,11 +141,11 @@
141
  <span class="tooltip-text">Number of complete passes through the dataset. More epochs improves learning but increases privacy budget consumption.</span>
142
  </span>
143
  </label>
144
- <input type="range" id="epochs" class="parameter-slider" min="1" max="20" step="1" value="5">
145
  <div class="slider-display">
146
  <span>1</span>
147
- <span id="epochs-value">5</span>
148
- <span>20</span>
149
  </div>
150
  </div>
151
 
@@ -214,7 +214,7 @@
214
  <div id="training-status" class="status-badge" style="display: none;">
215
  <span class="pulse"></span>
216
  <span style="font-weight: 500; color: #4caf50;">Training in progress</span>
217
- <span style="margin-left: auto; font-weight: 500;">Epoch: <span id="current-epoch">1</span> / <span id="total-epochs">5</span></span>
218
  </div>
219
  </div>
220
 
 
141
  <span class="tooltip-text">Number of complete passes through the dataset. More epochs improves learning but increases privacy budget consumption.</span>
142
  </span>
143
  </label>
144
+ <input type="range" id="epochs" class="parameter-slider" min="1" max="50" step="1" value="30">
145
  <div class="slider-display">
146
  <span>1</span>
147
+ <span id="epochs-value">30</span>
148
+ <span>50</span>
149
  </div>
150
  </div>
151
 
 
214
  <div id="training-status" class="status-badge" style="display: none;">
215
  <span class="pulse"></span>
216
  <span style="font-weight: 500; color: #4caf50;">Training in progress</span>
217
+ <span style="margin-left: auto; font-weight: 500;">Epoch: <span id="current-epoch">1</span> / <span id="total-epochs">30</span></span>
218
  </div>
219
  </div>
220