andras/bipolaroid
Watch
1
0
Fork
You've already forked bipolaroid
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update training

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-27 22:30:27 +01:00
commit 7863611f86
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
1 changed files with 95 additions and 135 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/training/random_hparam_search.py
View file

@ -36,16 +36,10 @@ def random_hparam_search(
log_dir = tensorboard_path / get_next_run_name(tensorboard_path)
try:
train_data_loader = get_data_loader(
train_data_paths, **current_hyperparameters
)
test_data_loader = get_data_loader(
test_data_paths, **current_hyperparameters
)
model = train(
hyperparameters=current_hyperparameters,
train_data_loader=train_data_loader,
test_data_loader=test_data_loader,
train_data_paths=train_data_paths,
test_data_paths=test_data_paths,
max_duration=timedelta(hours=timeout_hours),
log_dir=log_dir,
use_tqdm=False,