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

Update train.py

This commit is contained in:
Glenn Jocher 2019-03-07 13:28:54 +01:00 committed by GitHub
parent 473eb8d0c9
commit ff9d343019
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -87,7 +87,7 @@ def train(
t0 = time.time() t0 = time.time()
model_info(model) model_info(model)
n_burnin = min(round(dataloader.nB / 5), 1000) # number of burn-in batches n_burnin = min(round(dataloader.nB / 5 + 1), 1000) # number of burn-in batches
for epoch in range(epochs): for epoch in range(epochs):
epoch += start_epoch epoch += start_epoch