Skip to content

Commit f1fa332

Browse files
update frontend fields
1 parent 3bbc48f commit f1fa332

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

  • nebula/frontend/static/js/deployment

nebula/frontend/static/js/deployment/main.js

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ const DeploymentManager = (function() {
204204
datasetSelect.innerHTML = "";
205205

206206
// Add dataset options
207-
const datasets = ['MNIST', 'FashionMNIST', 'CIFAR10'];
207+
const datasets = ['MNIST', 'FashionMNIST', 'EMNIST', 'CIFAR10', 'CIFAR100'];
208208
datasets.forEach(dataset => {
209209
const option = document.createElement("option");
210210
option.value = dataset;
@@ -239,9 +239,12 @@ const DeploymentManager = (function() {
239239
switch(dataset.toLowerCase()) {
240240
case 'mnist':
241241
case 'fashionmnist':
242+
case 'emnist':
242243
return ['MLP', 'CNN'];
243244
case 'cifar10':
244-
return ['CNN', 'ResNet18'];
245+
return ['CNN', 'ResNet9', 'fastermobilenet', 'simplemobilenet', 'CNNv2', 'CNNv3'];
246+
case 'cifar100':
247+
return ['CNN'];
245248
default:
246249
return ['MLP', 'CNN'];
247250
}

0 commit comments

Comments
 (0)