mirror of
https://github.com/coqui-ai/TTS.git
synced 2025-12-25 20:59:48 +01:00
add RADAM reference
This commit is contained in:
@@ -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