Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-12-15 20:52:35 +01:00
parent 21ab0c76fd
commit 900851200e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -153,8 +153,8 @@ def train(
loss = model(imgs.to(device), targets, batch_report=report, var=var)
loss.backward()
accumulated_batches = 4 # accumulate gradient for 4 batches before optimizing
if ((i + 1) % accumulated_batches == 0) or (i == len(dataloader) - 1):
# accumulated_batches = 1 # accumulate gradient for 4 batches before optimizing
# if ((i + 1) % accumulated_batches == 0) or (i == len(dataloader) - 1):
optimizer.step()
optimizer.zero_grad()