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

updates

This commit is contained in:
Glenn Jocher 2019-09-17 15:16:29 +02:00
parent 425ed4b84b
commit 78bb153d7d
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -430,8 +430,8 @@ if __name__ == '__main__':
hyp[k] = x[i + 7]
# Mutate
init_seeds(seed=int(time.time()))
s = [.20, .20, .20, .20, .20, .20, .20, .00, .02, .20, .20, .20, .20, .20, .20, .20, .20] # sigmas
np.random.seed(int(time.time()))
s = [.1, .1, .1, .1, .1, .1, .1, .0, .02, .2, .2, .2, .2, .2, .2, .2, .2] # sigmas
for i, k in enumerate(hyp.keys()):
x = (np.random.randn(1) * s[i] + 1) ** 2.0 # plt.hist(x.ravel(), 300)
hyp[k] *= float(x) # vary by sigmas

1
utils/torch_utils.py
View File

@ -4,7 +4,6 @@ import torch
def init_seeds(seed=0):
torch.cuda.empty_cache()
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)