|
|
@ -1,4 +1,4 @@ |
|
|
import subprocess |
|
|
import subprocess # noqa: F401 |
|
|
import sys |
|
|
import sys |
|
|
from random import sample, seed |
|
|
from random import sample, seed |
|
|
|
|
|
|
|
|
@ -8,7 +8,7 @@ from lightning_sdk import Machine, Studio # noqa: F401 |
|
|
# consistency of randomly sampled experiments. |
|
|
# consistency of randomly sampled experiments. |
|
|
seed(19920921) |
|
|
seed(19920921) |
|
|
|
|
|
|
|
|
NUM_JOBS = 5 |
|
|
NUM_JOBS = 1 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Define the ranges or sets of values for each hyperparameter |
|
|
# Define the ranges or sets of values for each hyperparameter |
|
|
@ -25,7 +25,7 @@ depths = [1, 2, 4, 8, 16] |
|
|
|
|
|
|
|
|
batch_size_values = [256] |
|
|
batch_size_values = [256] |
|
|
max_epochs_values = [420] # at 12 fps, around 35s |
|
|
max_epochs_values = [420] # at 12 fps, around 35s |
|
|
seeds = list(range(21, 1992)) |
|
|
seeds = list(range(20, 1992)) |
|
|
optimizers = [ |
|
|
optimizers = [ |
|
|
# "Adagrad", |
|
|
# "Adagrad", |
|
|
"Adam", |
|
|
"Adam", |
|
|
|
Michael Pilosov, PhD
9 months ago