Spaces:
Running
Running
update
Browse files
examples/conv_tasnet/step_2_train_model.py
CHANGED
@@ -240,7 +240,6 @@ def main():
|
|
240 |
total_lds_loss = 0.
|
241 |
total_batches = 0.
|
242 |
progress_bar = tqdm(
|
243 |
-
total=len(train_data_loader),
|
244 |
desc="Training; epoch: {}".format(idx_epoch),
|
245 |
)
|
246 |
for batch in train_data_loader:
|
@@ -306,7 +305,6 @@ def main():
|
|
306 |
total_batches = 0.
|
307 |
|
308 |
progress_bar = tqdm(
|
309 |
-
total=len(valid_data_loader),
|
310 |
desc="Evaluation; epoch: {}".format(idx_epoch),
|
311 |
)
|
312 |
with torch.no_grad():
|
|
|
240 |
total_lds_loss = 0.
|
241 |
total_batches = 0.
|
242 |
progress_bar = tqdm(
|
|
|
243 |
desc="Training; epoch: {}".format(idx_epoch),
|
244 |
)
|
245 |
for batch in train_data_loader:
|
|
|
305 |
total_batches = 0.
|
306 |
|
307 |
progress_bar = tqdm(
|
|
|
308 |
desc="Evaluation; epoch: {}".format(idx_epoch),
|
309 |
)
|
310 |
with torch.no_grad():
|