diff --git a/userbenchmark/optim/run.py b/userbenchmark/optim/run.py index 9aef7cbcf9..0d11b956aa 100644 --- a/userbenchmark/optim/run.py +++ b/userbenchmark/optim/run.py @@ -60,7 +60,7 @@ def get_unstable_models() -> Set[str]: OPTIM_NAMES = [o.__name__ for o in [Adadelta, Adagrad, Adam, AdamW, Adamax, ASGD, SGD, RAdam, Rprop, RMSprop, NAdam, SparseAdam]] -FUNC_STRS = ['pt2_' , ''] +FUNC_STRS = [''] OPTIMIZERS = [ (Adadelta, {}),