terminator.utils.model.optim

Optimizers for TERMinator

Currently only includes the Noam optimizer, based on https://github.com/jingraham/neurips19-graph-protein-design

Functions

get_std_opt(parameters, d_model[, ...])

Classes

NoamOpt(model_size, factor, warmup, optimizer)

Optim wrapper that implements rate.