ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update examples to ClearML

Browse Source
This commit is contained in:
allegroai
2020-12-30 16:53:19 +02:00
parent 0bb590b5da
commit 4c8a24f1ff
17 changed files with 86 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
examples/frameworks/pytorch/notebooks/image/hyperparameter_search.ipynb
View File

@@ -7,12 +7,12 @@
"outputs": [],
"source": [
"# execute this in command line on all machines to be used as workers before initiating the hyperparamer search \n",
"# ! pip install -U trains-agent==0.15.0\n",
"# ! trains-agent daemon --queue default\n",
"# ! pip install -U clearml-agent==0.15.0\n",
"# ! clearml-agent daemon --queue default\n",
"\n",
"# pip install with locked versions\n",
"! pip install -U pandas==1.0.3\n",
"! pip install -U trains>=0.16.2\n",
"! pip install -U clearml>=0.16.2\n",
"! pip install -U optuna==2.0.0"
]
},
@@ -22,11 +22,11 @@
"metadata": {},
"outputs": [],
"source": [
"from trains.automation import UniformParameterRange, UniformIntegerParameterRange\n",
"from trains.automation import HyperParameterOptimizer\n",
"from trains.automation.optuna import OptimizerOptuna\n",
"from clearml.automation import UniformParameterRange, UniformIntegerParameterRange\n",
"from clearml.automation import HyperParameterOptimizer\n",
"from clearml.automation.optuna import OptimizerOptuna\n",
"\n",
"from trains import Task"
"from clearml import Task"
]
},
{
@@ -72,7 +72,7 @@
" objective_metric_series='total',\n",
" objective_metric_sign='max', # maximize or minimize the objective metric\n",
"\n",
" # setting optimizer - trains supports GridSearch, RandomSearch, OptimizerBOHB and OptimizerOptuna\n",
" # setting optimizer - clearml supports GridSearch, RandomSearch, OptimizerBOHB and OptimizerOptuna\n",
" optimizer_class=OptimizerOptuna,\n",
" \n",
" # Configuring optimization parameters\n",

6
examples/frameworks/pytorch/notebooks/image/image_classification_CIFAR10.ipynb
View File

@@ -15,7 +15,7 @@
"! pip install -U torch==1.5.1\n",
"! pip install -U torchvision==0.6.1\n",
"! pip install -U numpy==1.18.4\n",
"! pip install -U trains>=0.16.1\n",
"! pip install -U clearml>=0.16.1\n",
"! pip install -U tensorboard==2.2.1"
]
},
@@ -36,7 +36,7 @@
"import torchvision.datasets as datasets\n",
"import torchvision.transforms as transforms\n",
"\n",
"from trains import Task"
"from clearml import Task"
]
},
{
@@ -47,7 +47,7 @@
"source": [
"task = Task.init(project_name='Image Example', task_name='image classification CIFAR10')\n",
"configuration_dict = {'number_of_epochs': 3, 'batch_size': 4, 'dropout': 0.25, 'base_lr': 0.001}\n",
"configuration_dict = task.connect(configuration_dict) # enabling configuration override by trains\n",
"configuration_dict = task.connect(configuration_dict) # enabling configuration override by clearml\n",
"print(configuration_dict) # printing actual configuration (after override in remote mode)"
]
},