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

updates

This commit is contained in:
Glenn Jocher 2019-03-05 17:14:40 +01:00
parent a2ad00d6fc
commit 6fb14fc903
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()
model_info(model)
n_burnin = min(dataloader.nB / 5, 1000) # number of burn-in batches
n_burnin = min(round(dataloader.nB / 5), 1000) # number of burn-in batches
for epoch in range(epochs):
epoch += start_epoch