From 78bb153d7d5bdc9782b0e2ec8be391a33380614b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 17 Sep 2019 15:16:29 +0200 Subject: [PATCH] updates --- train.py | 4 ++-- utils/torch_utils.py | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index c94abb20..465a4118 100644 --- a/train.py +++ b/train.py @@ -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 diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 11b3504e..aa42cb0f 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -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)