From b773e2898f0c62091687001ce57f121716244077 Mon Sep 17 00:00:00 2001 From: Chi Wang Date: Mon, 6 Dec 2021 14:55:05 -0800 Subject: [PATCH] Update flaml_pytorch_cifar10.ipynb training_function -> train_cifar --- notebook/flaml_pytorch_cifar10.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebook/flaml_pytorch_cifar10.ipynb b/notebook/flaml_pytorch_cifar10.ipynb index 7e096607b8..aac0cdbc3d 100644 --- a/notebook/flaml_pytorch_cifar10.ipynb +++ b/notebook/flaml_pytorch_cifar10.ipynb @@ -315,7 +315,7 @@ " low_cost_partial_config={\"num_epochs\": 1},\n", " max_resource=max_num_epoch,\n", " min_resource=1,\n", - " scheduler=\"asha\", # need to use tune.report to report intermediate results in training_function \n", + " scheduler=\"asha\", # need to use tune.report to report intermediate results in train_cifar \n", " resources_per_trial={\"cpu\": 1, \"gpu\": gpus_per_trial},\n", " local_dir='logs/',\n", " num_samples=num_samples,\n",