Fix get_instance_metadata for the mnist model in the dev harness (#77)

Co-authored-by: Nicholas King <v-nicki@microsoft.com>
This commit is contained in:
Nicholas King 2020-11-20 14:18:51 -08:00 коммит произвёл GitHub
Родитель c8ebe6adb7
Коммит ca68abbcfe
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -194,8 +194,8 @@ def mnist_sweep():
return send_file(img_bytes, mimetype='image/png')
def get_instance_label(instance_id):
label = data_loader[instance_id][2].item()
return {"label": label}
label = dataset[instance_id][2].item()
return f'{label}'
CompatibilityAnalysis(sweeps_folder, n_epochs, h1, h2, train_loader, test_loader,
batch_size_train, batch_size_test,