wangtao
/
autogen
mirror of https://github.com/microsoft/autogen.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update flaml_pytorch_cifar10.ipynb 

training_function -> train_cifar
This commit is contained in:
Chi Wang 2021-12-06 14:55:05 -08:00
parent dd60dbc5eb
commit b773e2898f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
notebook/flaml_pytorch_cifar10.ipynb
View File

@ -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",