Skip to content

Instantly share code, notes, and snippets.

@amogkam

amogkam/pbt.py

Created August 17, 2020 17:17
Show Gist options
  • Save amogkam/f5ea567fa6005d93370322c488731235 to your computer and use it in GitHub Desktop.
Save amogkam/f5ea567fa6005d93370322c488731235 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pbt.py
{
"per_gpu_batch_size": [16, 32, 64],
"weight_decay": (0, 0.3),
"learning_rate": (1e-5, 5e-5),
"num_epochs": [2, 3, 4, 5]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment