mirror of
https://github.com/wassname/TTS.git
synced 2026-10-25 12:10:21 +08:00
add RADAM reference
This commit is contained in:
1 parent
529348d6dc
commit
dc69074a56
1 file changed
+1
@@ -3,6 +3,7 @@ import torch
|
||||
from torch.optim.optimizer import Optimizer
|
||||
|
||||
|
||||
# adapted from https://github.com/LiyuanLucasLiu/RAdam
|
||||
class RAdam(Optimizer):
|
||||
|
||||
def __init__(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8, weight_decay=0):
|
||||
|
||||
Reference in new issue
Block a user