Skip to content

Instantly share code, notes, and snippets.

@ryujaehun

ryujaehun/LUT_Parsing_CUDA.ipynb

Created April 8, 2020 06:13
Show Gist options
  • Save ryujaehun/702b52deb3e792b7f802ba8b5f5ba79b to your computer and use it in GitHub Desktop.
Save ryujaehun/702b52deb3e792b7f802ba8b5f5ba79b to your computer and use it in GitHub Desktop.
Download ZIP
LUT_Parsing_CUDA
Display the source blob
Display the rendered blob
Raw
LUT_Parsing_CUDA.ipynb
{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"import os,glob\n",
"import pandas as pd\n",
"########################\n",
"import sys\n"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
"tuners=['grid','xgb','rl']"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"['cuda/2020-03-24_00:43:08.837939/resnet-34.log',\n",
" 'cuda/2020-03-24_00:42:51.813383/resnet-34.log',\n",
" 'cuda/2020-03-24_00:42:34.788976/resnet-34.log']"
]
},
"execution_count": 3,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"logs=glob.glob('cuda/**/resnet-34.log',recursive=True)\n",
"logs"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:16: SettingWithCopyWarning: \n",
"A value is trying to be set on a copy of a slice from a DataFrame.\n",
"Try using .loc[row_indexer,col_indexer] = value instead\n",
"\n",
"See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n",
" app.launch_new_instance()\n"
]
}
],
"source": [
"results=dict()\n",
"for tuner,logname in zip(tuners,logs):\n",
" key=tuner+'_'+logname.split('/')[-1]\n",
" log_df=pd.read_json(logname, lines=True)\n",
" log_df['index']=log_df['input'].apply(lambda x:str(x[2]))\n",
" log_df['score']=log_df['result'].apply(lambda x:float(x[0][0]))\n",
" #log_df['score2']=log_df['r'].apply(lambda x:str(x[2]))\n",
" log_df['config']=log_df['config'].apply(lambda x:x['entity'])\n",
" score1=[]\n",
" score2=[]\n",
" score3=[]\n",
" score4=[]\n",
" score5=[]\n",
" log_group_df=log_df.groupby('index')\n",
" for _,group in log_group_df:\n",
" group['rank']=group['score'].rank(method='min',ascending=True)\n",
" score1.append(group[group['rank']==1.0].index[0])\n",
" score2.append(group[group['rank']==2.0].index[0])\n",
" score3.append(group[group['rank']==3.0].index[0])\n",
" score4.append(group[group['rank']==4.0].index[0])\n",
" score5.append(group[group['rank']==5.0].index[0])\n",
" results[key+'_score1']=log_df.iloc[score1]\n",
" results[key+'_score2']=log_df.iloc[score2]\n",
" results[key+'_score3']=log_df.iloc[score3]\n",
" results[key+'_score4']=log_df.iloc[score4]\n",
" results[key+'_score5']=log_df.iloc[score5]"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
"for key in results.keys():\n",
" try:\n",
" results[key]=results[key].drop(['input','result','version','tvm_version'],axis=1)\n",
" results[key].sort_index(inplace=True)\n",
" except Exception as e:\n",
" print(e)\n",
" for config in results[key]['config']:\n",
" for s in config:\n",
" if s[0]=='tile_b':\n",
" results[key]['tile_b']= str(s[2])\n",
" elif s[0]=='tile_f':\n",
" results[key]['tile_f']= str(s[2])\n",
" elif s[0]=='tile_x':\n",
" results[key]['tile_x']= str(s[2])\n",
" elif s[0]=='tile_y':\n",
" results[key]['tile_y']= str(s[2])\n",
" elif s[0]=='tile_rc':\n",
" results[key]['tile_rc']= str(s[2])\n",
" elif s[0]=='tile_rx':\n",
" results[key]['tile_rx']= str(s[2])\n",
" elif s[0]=='tile_ry':\n",
" results[key]['tile_ry']= str(s[2])\n",
" elif s[0]=='auto_unroll_max_step':\n",
" results[key]['auto_unroll_max_step']= str(s[2])\n",
" elif s[0]=='unroll_explicit':\n",
" results[key]['unroll_explicit']= str(s[2])\n",
" else:\n",
" print (\"No Key\")\n",
" \n",
"results[key].drop('config',axis=1,inplace=True)"
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" <th>tile_b</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>1480</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000034</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>5520</th>\n",
" <td>[[tile_f, sp, [-1, 8, 4, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000023</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>10260</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000022</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>19300</th>\n",
" <td>[[tile_f, sp, [-1, 2, 16, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000026</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>23780</th>\n",
" <td>[[tile_f, sp, [-1, 8, 4, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000200</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>33600</th>\n",
" <td>[[tile_f, sp, [-1, 4, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000105</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>38520</th>\n",
" <td>[[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000103</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>44560</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000106</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 8, 4]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>128</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"1480 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"5520 [[tile_f, sp, [-1, 8, 4, 1]], [tile_y, sp, [-1... \n",
"10260 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"19300 [[tile_f, sp, [-1, 2, 16, 2]], [tile_y, sp, [-... \n",
"23780 [[tile_f, sp, [-1, 8, 4, 1]], [tile_y, sp, [-1... \n",
"33600 [[tile_f, sp, [-1, 4, 8, 1]], [tile_y, sp, [-1... \n",
"38520 [[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1... \n",
"44560 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"\n",
" index score \\\n",
"1480 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000034 \n",
"5520 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000023 \n",
"10260 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000022 \n",
"19300 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000026 \n",
"23780 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000200 \n",
"33600 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000105 \n",
"38520 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000103 \n",
"44560 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000106 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"1480 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"5520 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"10260 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"19300 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"23780 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"33600 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"38520 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"44560 [-1, 8, 8, 1] [-1, 4, 8, 4] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit tile_b \n",
"1480 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"5520 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"10260 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"19300 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"23780 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"33600 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"38520 [-1, 3] 128 0 [-1, 1, 1, 1] \n",
"44560 [-1, 3] 128 0 [-1, 1, 1, 1] "
]
},
"execution_count": 6,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['grid_resnet-34.log_score1']"
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" <th>tile_b</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>400</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000038</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>6542</th>\n",
" <td>[[tile_f, sp, [-1, 2, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000023</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>10781</th>\n",
" <td>[[tile_f, sp, [-1, 2, 16, 4]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000023</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>17881</th>\n",
" <td>[[tile_f, sp, [-1, 1, 4, 4]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000031</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>21602</th>\n",
" <td>[[tile_f, sp, [-1, 4, 2, 4]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000210</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>28140</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000111</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>38206</th>\n",
" <td>[[tile_f, sp, [-1, 2, 8, 2]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000116</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>40905</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000107</td>\n",
" <td>[-1, 2, 8, 2]</td>\n",
" <td>[-1, 8, 8, 2]</td>\n",
" <td>[-1, 7, 16, 1]</td>\n",
" <td>[-1, 32]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>0</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"400 [[tile_f, sp, [-1, 4, 16, 2]], [tile_y, sp, [-... \n",
"6542 [[tile_f, sp, [-1, 2, 64, 1]], [tile_y, sp, [-... \n",
"10781 [[tile_f, sp, [-1, 2, 16, 4]], [tile_y, sp, [-... \n",
"17881 [[tile_f, sp, [-1, 1, 4, 4]], [tile_y, sp, [-1... \n",
"21602 [[tile_f, sp, [-1, 4, 2, 4]], [tile_y, sp, [-1... \n",
"28140 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"38206 [[tile_f, sp, [-1, 2, 8, 2]], [tile_y, sp, [-1... \n",
"40905 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"\n",
" index score \\\n",
"400 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000038 \n",
"6542 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000023 \n",
"10781 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000023 \n",
"17881 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000031 \n",
"21602 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000210 \n",
"28140 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000111 \n",
"38206 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000116 \n",
"40905 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000107 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"400 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"6542 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"10781 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"17881 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"21602 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"28140 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"38206 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"40905 [-1, 2, 8, 2] [-1, 8, 8, 2] [-1, 7, 16, 1] [-1, 32] [-1, 3] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit tile_b \n",
"400 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"6542 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"10781 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"17881 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"21602 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"28140 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"38206 [-1, 3] 0 1 [-1, 1, 1, 1] \n",
"40905 [-1, 3] 0 1 [-1, 1, 1, 1] "
]
},
"execution_count": 7,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['grid_resnet-34.log_score2']"
]
},
{
"cell_type": "code",
"execution_count": 8,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" <th>tile_b</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>3081</th>\n",
" <td>[[tile_f, sp, [-1, 2, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000039</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>5401</th>\n",
" <td>[[tile_f, sp, [-1, 8, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000024</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>14420</th>\n",
" <td>[[tile_f, sp, [-1, 1, 32, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000025</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>16680</th>\n",
" <td>[[tile_f, sp, [-1, 8, 4, 2]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000032</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>24422</th>\n",
" <td>[[tile_f, sp, [-1, 16, 2, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000231</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>29040</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000111</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>36991</th>\n",
" <td>[[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000118</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>41220</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000108</td>\n",
" <td>[-1, 8, 8, 1]</td>\n",
" <td>[-1, 4, 16, 2]</td>\n",
" <td>[-1, 4, 14, 1]</td>\n",
" <td>[-1, 8]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"3081 [[tile_f, sp, [-1, 2, 16, 1]], [tile_y, sp, [-... \n",
"5401 [[tile_f, sp, [-1, 8, 32, 1]], [tile_y, sp, [-... \n",
"14420 [[tile_f, sp, [-1, 1, 32, 2]], [tile_y, sp, [-... \n",
"16680 [[tile_f, sp, [-1, 8, 4, 2]], [tile_y, sp, [-1... \n",
"24422 [[tile_f, sp, [-1, 16, 2, 2]], [tile_y, sp, [-... \n",
"29040 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"36991 [[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1... \n",
"41220 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"\n",
" index score \\\n",
"3081 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000039 \n",
"5401 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000024 \n",
"14420 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000025 \n",
"16680 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000032 \n",
"24422 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000231 \n",
"29040 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000111 \n",
"36991 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000118 \n",
"41220 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000108 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"3081 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"5401 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"14420 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"16680 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"24422 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"29040 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"36991 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"41220 [-1, 8, 8, 1] [-1, 4, 16, 2] [-1, 4, 14, 1] [-1, 8] [-1, 3] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit tile_b \n",
"3081 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"5401 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"14420 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"16680 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"24422 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"29040 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"36991 [-1, 3] 1500 0 [-1, 1, 1, 1] \n",
"41220 [-1, 3] 1500 0 [-1, 1, 1, 1] "
]
},
"execution_count": 8,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['grid_resnet-34.log_score3']"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" <th>tile_b</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>963</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000031</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>5585</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000016</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>10463</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000017</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>16044</th>\n",
" <td>[[tile_f, sp, [-1, 2, 8, 4]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000021</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>20591</th>\n",
" <td>[[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000123</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>31800</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000097</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>35001</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000081</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>40091</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000095</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>50400</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000095</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>58336</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000080</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>66703</th>\n",
" <td>[[tile_f, sp, [-1, 4, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000167</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>70780</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 512, 7, 7], 'float32'], ['TENS...</td>\n",
" <td>0.000081</td>\n",
" <td>[-1, 4, 8, 1]</td>\n",
" <td>[-1, 2, 8, 4]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"963 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"5585 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"10463 [[tile_f, sp, [-1, 4, 16, 2]], [tile_y, sp, [-... \n",
"16044 [[tile_f, sp, [-1, 2, 8, 4]], [tile_y, sp, [-1... \n",
"20591 [[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1... \n",
"31800 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"35001 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"40091 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"50400 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"58336 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"66703 [[tile_f, sp, [-1, 4, 8, 1]], [tile_y, sp, [-1... \n",
"70780 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"\n",
" index score \\\n",
"963 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000031 \n",
"5585 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000016 \n",
"10463 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000017 \n",
"16044 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000021 \n",
"20591 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000123 \n",
"31800 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000097 \n",
"35001 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000081 \n",
"40091 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000095 \n",
"50400 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000095 \n",
"58336 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000080 \n",
"66703 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000167 \n",
"70780 [['TENSOR', [4, 512, 7, 7], 'float32'], ['TENS... 0.000081 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"963 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"5585 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"10463 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"16044 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"20591 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"31800 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"35001 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"40091 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"50400 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"58336 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"66703 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"70780 [-1, 4, 8, 1] [-1, 2, 8, 4] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit tile_b \n",
"963 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"5585 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"10463 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"16044 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"20591 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"31800 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"35001 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"40091 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"50400 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"58336 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"66703 [-1, 3] 1500 1 [-1, 1, 1, 1] \n",
"70780 [-1, 3] 1500 1 [-1, 1, 1, 1] "
]
},
"execution_count": 9,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['xgb_resnet-34.log_score1']"
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" <th>tile_b</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>2005</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000032</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>5571</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000016</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>10363</th>\n",
" <td>[[tile_f, sp, [-1, 16, 8, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000017</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>15907</th>\n",
" <td>[[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000021</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>20592</th>\n",
" <td>[[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000123</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>31861</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000097</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>35005</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000081</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>40139</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000096</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>50337</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000096</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>58337</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000080</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>66680</th>\n",
" <td>[[tile_f, sp, [-1, 2, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000168</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>70842</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 512, 7, 7], 'float32'], ['TENS...</td>\n",
" <td>0.000081</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 4, 16, 1]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>0</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"2005 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"5571 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"10363 [[tile_f, sp, [-1, 16, 8, 1]], [tile_y, sp, [-... \n",
"15907 [[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1... \n",
"20592 [[tile_f, sp, [-1, 8, 8, 1]], [tile_y, sp, [-1... \n",
"31861 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"35005 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"40139 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"50337 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"58337 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"66680 [[tile_f, sp, [-1, 2, 16, 1]], [tile_y, sp, [-... \n",
"70842 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"\n",
" index score \\\n",
"2005 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000032 \n",
"5571 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000016 \n",
"10363 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000017 \n",
"15907 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000021 \n",
"20592 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000123 \n",
"31861 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000097 \n",
"35005 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000081 \n",
"40139 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000096 \n",
"50337 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000096 \n",
"58337 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000080 \n",
"66680 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000168 \n",
"70842 [['TENSOR', [4, 512, 7, 7], 'float32'], ['TENS... 0.000081 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"2005 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"5571 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"10363 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"15907 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"20592 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"31861 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"35005 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"40139 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"50337 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"58337 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"66680 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"70842 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 4, 16, 1] [-1, 16] [-1, 3] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit tile_b \n",
"2005 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"5571 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"10363 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"15907 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"20592 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"31861 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"35005 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"40139 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"50337 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"58337 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"66680 [-1, 2] 1500 0 [-1, 1, 1, 1] \n",
"70842 [-1, 2] 1500 0 [-1, 1, 1, 1] "
]
},
"execution_count": 10,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['xgb_resnet-34.log_score2']"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" <th>tile_b</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>2002</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000032</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>5582</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000017</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>10359</th>\n",
" <td>[[tile_f, sp, [-1, 16, 8, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000017</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>15851</th>\n",
" <td>[[tile_f, sp, [-1, 4, 8, 2]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000021</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>23998</th>\n",
" <td>[[tile_f, sp, [-1, 1, 8, 8]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000125</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>31226</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000098</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>35002</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000081</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>40145</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000096</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>50292</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000097</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>58375</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000080</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>68102</th>\n",
" <td>[[tile_f, sp, [-1, 2, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000168</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" <tr>\n",
" <th>70522</th>\n",
" <td>[[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1...</td>\n",
" <td>[['TENSOR', [4, 512, 7, 7], 'float32'], ['TENS...</td>\n",
" <td>0.000082</td>\n",
" <td>[-1, 2, 16, 1]</td>\n",
" <td>[-1, 1, 8, 8]</td>\n",
" <td>[-1, 2, 16, 2]</td>\n",
" <td>[-1, 16]</td>\n",
" <td>[-1, 3]</td>\n",
" <td>[-1, 2]</td>\n",
" <td>1500</td>\n",
" <td>1</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"2002 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"5582 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"10359 [[tile_f, sp, [-1, 16, 8, 1]], [tile_y, sp, [-... \n",
"15851 [[tile_f, sp, [-1, 4, 8, 2]], [tile_y, sp, [-1... \n",
"23998 [[tile_f, sp, [-1, 1, 8, 8]], [tile_y, sp, [-1... \n",
"31226 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"35002 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"40145 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"50292 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"58375 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"68102 [[tile_f, sp, [-1, 2, 16, 1]], [tile_y, sp, [-... \n",
"70522 [[tile_b, sp, [-1, 1, 1, 1]], [tile_y, sp, [-1... \n",
"\n",
" index score \\\n",
"2002 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000032 \n",
"5582 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000017 \n",
"10359 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000017 \n",
"15851 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000021 \n",
"23998 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000125 \n",
"31226 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000098 \n",
"35002 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000081 \n",
"40145 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000096 \n",
"50292 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000097 \n",
"58375 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000080 \n",
"68102 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000168 \n",
"70522 [['TENSOR', [4, 512, 7, 7], 'float32'], ['TENS... 0.000082 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"2002 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"5582 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"10359 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"15851 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"23998 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"31226 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"35002 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"40145 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"50292 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"58375 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"68102 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"70522 [-1, 2, 16, 1] [-1, 1, 8, 8] [-1, 2, 16, 2] [-1, 16] [-1, 3] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit tile_b \n",
"2002 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"5582 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"10359 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"15851 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"23998 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"31226 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"35002 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"40145 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"50292 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"58375 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"68102 [-1, 2] 1500 1 [-1, 1, 1, 1] \n",
"70522 [-1, 2] 1500 1 [-1, 1, 1, 1] "
]
},
"execution_count": 11,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['xgb_resnet-34.log_score3']"
]
},
{
"cell_type": "code",
"execution_count": 12,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>20476</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000043</td>\n",
" <td>[-1, 2, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>98542</th>\n",
" <td>[[tile_f, sp, [-1, 2, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000024</td>\n",
" <td>[-1, 2, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>118401</th>\n",
" <td>[[tile_f, sp, [-1, 8, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000045</td>\n",
" <td>[-1, 2, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>160446</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000039</td>\n",
" <td>[-1, 2, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>200508</th>\n",
" <td>[[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000867</td>\n",
" <td>[-1, 2, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"20476 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"98542 [[tile_f, sp, [-1, 2, 64, 1]], [tile_y, sp, [-... \n",
"118401 [[tile_f, sp, [-1, 8, 16, 1]], [tile_y, sp, [-... \n",
"160446 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"200508 [[tile_f, sp, [-1, 2, 32, 1]], [tile_y, sp, [-... \n",
"\n",
" index score \\\n",
"20476 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000043 \n",
"98542 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000024 \n",
"118401 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000045 \n",
"160446 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000039 \n",
"200508 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000867 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"20476 [-1, 2, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"98542 [-1, 2, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"118401 [-1, 2, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"160446 [-1, 2, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"200508 [-1, 2, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit \n",
"20476 [-1, 1] 0 0 \n",
"98542 [-1, 1] 0 0 \n",
"118401 [-1, 1] 0 0 \n",
"160446 [-1, 1] 0 0 \n",
"200508 [-1, 1] 0 0 "
]
},
"execution_count": 12,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['rl_resnet-34.log_score1']"
]
},
{
"cell_type": "code",
"execution_count": 13,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>15907</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000047</td>\n",
" <td>[-1, 1, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>98583</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000025</td>\n",
" <td>[-1, 1, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>118429</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 2]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000045</td>\n",
" <td>[-1, 1, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>163722</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000040</td>\n",
" <td>[-1, 1, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>200507</th>\n",
" <td>[[tile_f, sp, [-1, 1, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000959</td>\n",
" <td>[-1, 1, 32, 1]</td>\n",
" <td>[-1, 1, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"15907 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"98583 [[tile_f, sp, [-1, 1, 64, 2]], [tile_y, sp, [-... \n",
"118429 [[tile_f, sp, [-1, 4, 16, 2]], [tile_y, sp, [-... \n",
"163722 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"200507 [[tile_f, sp, [-1, 1, 32, 1]], [tile_y, sp, [-... \n",
"\n",
" index score \\\n",
"15907 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000047 \n",
"98583 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000025 \n",
"118429 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000045 \n",
"163722 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000040 \n",
"200507 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000959 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"15907 [-1, 1, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"98583 [-1, 1, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"118429 [-1, 1, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"163722 [-1, 1, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"200507 [-1, 1, 32, 1] [-1, 1, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit \n",
"15907 [-1, 1] 0 0 \n",
"98583 [-1, 1] 0 0 \n",
"118429 [-1, 1] 0 0 \n",
"163722 [-1, 1] 0 0 \n",
"200507 [-1, 1] 0 0 "
]
},
"execution_count": 13,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['rl_resnet-34.log_score2']"
]
},
{
"cell_type": "code",
"execution_count": 14,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<div>\n",
"<style scoped>\n",
" .dataframe tbody tr th:only-of-type {\n",
" vertical-align: middle;\n",
" }\n",
"\n",
" .dataframe tbody tr th {\n",
" vertical-align: top;\n",
" }\n",
"\n",
" .dataframe thead th {\n",
" text-align: right;\n",
" }\n",
"</style>\n",
"<table border=\"1\" class=\"dataframe\">\n",
" <thead>\n",
" <tr style=\"text-align: right;\">\n",
" <th></th>\n",
" <th>config</th>\n",
" <th>index</th>\n",
" <th>score</th>\n",
" <th>tile_f</th>\n",
" <th>tile_y</th>\n",
" <th>tile_x</th>\n",
" <th>tile_rc</th>\n",
" <th>tile_ry</th>\n",
" <th>tile_rx</th>\n",
" <th>auto_unroll_max_step</th>\n",
" <th>unroll_explicit</th>\n",
" </tr>\n",
" </thead>\n",
" <tbody>\n",
" <tr>\n",
" <th>15440</th>\n",
" <td>[[tile_f, sp, [-1, 2, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 256, 14, 14], 'float32'], ['TE...</td>\n",
" <td>0.000048</td>\n",
" <td>[-1, 1, 64, 1]</td>\n",
" <td>[-1, 2, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>85488</th>\n",
" <td>[[tile_f, sp, [-1, 8, 32, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 128, 28, 28], 'float32'], ['TE...</td>\n",
" <td>0.000026</td>\n",
" <td>[-1, 1, 64, 1]</td>\n",
" <td>[-1, 2, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>118469</th>\n",
" <td>[[tile_f, sp, [-1, 1, 16, 8]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000045</td>\n",
" <td>[-1, 1, 64, 1]</td>\n",
" <td>[-1, 2, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>161202</th>\n",
" <td>[[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN...</td>\n",
" <td>0.000040</td>\n",
" <td>[-1, 1, 64, 1]</td>\n",
" <td>[-1, 2, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>200606</th>\n",
" <td>[[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-...</td>\n",
" <td>[['TENSOR', [4, 3, 224, 224], 'float32'], ['TE...</td>\n",
" <td>0.000961</td>\n",
" <td>[-1, 1, 64, 1]</td>\n",
" <td>[-1, 2, 1, 1]</td>\n",
" <td>[-1, 1, 1, 32]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>[-1, 1]</td>\n",
" <td>0</td>\n",
" <td>0</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" config \\\n",
"15440 [[tile_f, sp, [-1, 2, 64, 1]], [tile_y, sp, [-... \n",
"85488 [[tile_f, sp, [-1, 8, 32, 1]], [tile_y, sp, [-... \n",
"118469 [[tile_f, sp, [-1, 1, 16, 8]], [tile_y, sp, [-... \n",
"161202 [[tile_f, sp, [-1, 4, 16, 1]], [tile_y, sp, [-... \n",
"200606 [[tile_f, sp, [-1, 1, 64, 1]], [tile_y, sp, [-... \n",
"\n",
" index score \\\n",
"15440 [['TENSOR', [4, 256, 14, 14], 'float32'], ['TE... 0.000048 \n",
"85488 [['TENSOR', [4, 128, 28, 28], 'float32'], ['TE... 0.000026 \n",
"118469 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000045 \n",
"161202 [['TENSOR', [4, 64, 56, 56], 'float32'], ['TEN... 0.000040 \n",
"200606 [['TENSOR', [4, 3, 224, 224], 'float32'], ['TE... 0.000961 \n",
"\n",
" tile_f tile_y tile_x tile_rc tile_ry \\\n",
"15440 [-1, 1, 64, 1] [-1, 2, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"85488 [-1, 1, 64, 1] [-1, 2, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"118469 [-1, 1, 64, 1] [-1, 2, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"161202 [-1, 1, 64, 1] [-1, 2, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"200606 [-1, 1, 64, 1] [-1, 2, 1, 1] [-1, 1, 1, 32] [-1, 1] [-1, 1] \n",
"\n",
" tile_rx auto_unroll_max_step unroll_explicit \n",
"15440 [-1, 1] 0 0 \n",
"85488 [-1, 1] 0 0 \n",
"118469 [-1, 1] 0 0 \n",
"161202 [-1, 1] 0 0 \n",
"200606 [-1, 1] 0 0 "
]
},
"execution_count": 14,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"results['rl_resnet-34.log_score3']"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.6.9"
}
},
"nbformat": 4,
"nbformat_minor": 4
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment