clean code
This commit is contained in:
parent
8ac3a94358
commit
077a947972
@ -77,7 +77,7 @@ class PyTorchModelTrainer:
|
|||||||
batch_size=self.batch_size,
|
batch_size=self.batch_size,
|
||||||
n_iters=self.max_iters
|
n_iters=self.max_iters
|
||||||
)
|
)
|
||||||
for epoch in range(1, epochs+1):
|
for epoch in range(1, epochs + 1):
|
||||||
# training
|
# training
|
||||||
losses = []
|
losses = []
|
||||||
for i, batch_data in enumerate(data_loaders_dictionary["train"]):
|
for i, batch_data in enumerate(data_loaders_dictionary["train"]):
|
||||||
|
Loading…
Reference in New Issue
Block a user