зеркало из https://github.com/mozilla/TTS.git
fix memory bloat on restore
This commit is contained in:
Родитель
40a533530b
Коммит
c466b9c1ff
2
train.py
2
train.py
|
@ -569,7 +569,7 @@ def main(args): # pylint: disable=redefined-outer-name
|
|||
pos_weight=torch.tensor(10)) if c.stopnet else None
|
||||
|
||||
if args.restore_path:
|
||||
checkpoint = torch.load(args.restore_path)
|
||||
checkpoint = torch.load(args.restore_path, map_location='cpu')
|
||||
try:
|
||||
# TODO: fix optimizer init, model.cuda() needs to be called before
|
||||
# optimizer restore
|
||||
|
|
Загрузка…
Ссылка в новой задаче