Skip to content

Commit

Permalink
add remaining models
Browse files Browse the repository at this point in the history
  • Loading branch information
@jmoralez
jmoralez committed Nov 7, 2023
1 parent e817979 commit e050a4d
Show file tree
Hide file tree
Showing 2 changed files with 84 additions and 16 deletions.
66 changes: 50 additions & 16 deletions nbs/models.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,7 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" backend='ray',\n",
" callbacks=None,\n",
" ):\n",
" \"\"\" Auto RNN\n",
" \n",
Expand Down Expand Up @@ -209,6 +210,7 @@
" gpus=gpus,\n",
" verbose=verbose,\n",
" backend=backend,\n",
" callbacks=callbacks,\n",
" )"
]
},
Expand Down Expand Up @@ -306,7 +308,8 @@
" cpus=cpu_count(),\n",
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
"\n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand All @@ -332,6 +335,7 @@
" gpus=gpus,\n",
" verbose=verbose,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -403,7 +407,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand All @@ -430,6 +435,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -500,7 +506,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand All @@ -527,6 +534,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -598,7 +606,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -627,6 +636,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -699,7 +709,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand All @@ -726,6 +737,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks,\n",
" )"
]
},
Expand Down Expand Up @@ -805,7 +817,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
"\n",
" # Define search space, input/output sizes \n",
" if config is None:\n",
Expand Down Expand Up @@ -834,6 +847,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -902,7 +916,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes \n",
" if config is None:\n",
Expand Down Expand Up @@ -931,6 +946,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1000,7 +1016,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1029,6 +1046,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1215,7 +1233,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1244,6 +1263,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1314,7 +1334,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1343,6 +1364,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1413,7 +1435,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1442,6 +1465,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1512,7 +1536,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1541,6 +1566,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1610,7 +1636,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes \n",
" if config is None:\n",
Expand Down Expand Up @@ -1639,6 +1666,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1711,7 +1739,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1740,6 +1769,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1819,7 +1849,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1848,6 +1879,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down Expand Up @@ -1924,7 +1956,8 @@
" gpus=torch.cuda.device_count(),\n",
" verbose=False,\n",
" alias=None,\n",
" backend='ray'):\n",
" backend='ray',\n",
" callbacks=None):\n",
" \n",
" # Define search space, input/output sizes\n",
" if config is None:\n",
Expand Down Expand Up @@ -1956,6 +1989,7 @@
" verbose=verbose,\n",
" alias=alias,\n",
" backend=backend,\n",
" callbacks=callbacks, \n",
" )"
]
},
Expand Down
Loading

0 comments on commit e050a4d

Please sign in to comment.