Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
executable file 15 lines (14 sloc) 348 Bytes
#!/usr/bin/env python
import wandb
from train_lib import train
if __name__ == '__main__':
sweep_config = {
'method': 'grid',
'parameters': {
'layers': {
'values': [32, 64, 96, 128, 256]
}
}
}
sweep_id = wandb.sweep(sweep_config)
wandb.agent(sweep_id, function=train)
You can’t perform that action at this time.