Searched refs:log_interval (Results 1 – 8 of 8) sorted by relevance
| /external/libopus/dnn/torch/osce/engine/ |
| D | vocoder_engine.py | 5 def train_one_epoch(model, criterion, optimizer, dataloader, device, scheduler, log_interval=10): argument
53 if i % log_interval == 0:
62 def evaluate(model, criterion, dataloader, device, log_interval=10): argument
94 if i % log_interval == 0:
|
| D | engine.py | 5 def train_one_epoch(model, criterion, optimizer, dataloader, device, scheduler, log_interval=10): argument
57 if i % log_interval == 0:
66 def evaluate(model, criterion, dataloader, device, log_interval=10): argument
96 if i % log_interval == 0:
|
| /external/libopus/dnn/torch/lpcnet/engine/ |
| D | lpcnet_engine.py | 34 def train_one_epoch(model, criterion, optimizer, dataloader, device, scheduler, log_interval=10): argument
86 if i % log_interval == 0:
95 def evaluate(model, criterion, dataloader, device, log_interval=10): argument
134 if i % log_interval == 0:
|
| /external/pytorch/torch/csrc/lazy/ |
| D | test_mnist.py | 46 def train(log_interval, model, device, train_loader, optimizer, epoch): argument
57 if batch_idx % log_interval == 0:
69 log_interval = 10 variable
92 train(log_interval, model, device, train_loader, optimizer, epoch)
|
| D | tutorial.md | 188 log_interval = 10
211 train(log_interval, model, device, train_loader, optimizer, epoch)
218 …ext iteration ...and then the one after next until we hit `if batch_idx % log_interval == 0:` where
226 def train(log_interval, model, device, train_loader, optimizer, epoch):
237 if batch_idx % log_interval == 0:
|
| /external/libopus/dnn/torch/rdovae/ |
| D | train_rdovae.py | 99 log_interval = 10 variable
257 if (i + 1) % log_interval == 0:
258 current_loss = (running_total_loss - previous_total_loss) / log_interval
|
| /external/libopus/dnn/torch/osce/ |
| D | adv_train_vocoder.py | 306 log_interval = 10 variable
413 if i % log_interval == 0:
|
| D | adv_train_model.py | 304 log_interval = 10 variable
422 if i % log_interval == 0:
|