Skip to content

Commit 9f325b1

Browse files
committed
fixup! Fix aux unit tests
1 parent fc09921 commit 9f325b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

TTS/bin/train_encoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def evaluation(model, criterion, data_loader, global_step):
125125

126126
def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader, global_step):
127127
model.train()
128-
best_loss = float("inf")
128+
best_loss = {"train_loss": None, "eval_loss": float("inf")}
129129
avg_loader_time = 0
130130
end_time = time.time()
131131
for epoch in range(c.epochs):

0 commit comments

Comments
 (0)