Skip to content
Snippets Groups Projects
Commit 999d6ff7 authored by tomrink's avatar tomrink
Browse files

minor

parent d5ce6f0f
No related branches found
No related tags found
No related merge requests found
...@@ -27,6 +27,8 @@ EARLY_STOP = True ...@@ -27,6 +27,8 @@ EARLY_STOP = True
TRIPLET = False TRIPLET = False
CONV3D = False CONV3D = False
NOISE_TRAINING = False
img_width = 16 img_width = 16
mean_std_file = homedir+'data/icing/mean_std_l1b_no_ice.pkl' mean_std_file = homedir+'data/icing/mean_std_l1b_no_ice.pkl'
...@@ -211,7 +213,10 @@ class IcingIntensityNN: ...@@ -211,7 +213,10 @@ class IcingIntensityNN:
data = [] data = []
for param in train_params: for param in train_params:
nda = h5f[param][nd_idxs, ] nda = h5f[param][nd_idxs, ]
nda = normalize(nda, param, mean_std_dct) if NOISE_TRAINING and is_training:
nda = normalize(nda, param, mean_std_dct, add_noise=True, noise_scale=0.01, seed=42)
else:
nda = normalize(nda, param, mean_std_dct)
data.append(nda) data.append(nda)
data = np.stack(data) data = np.stack(data)
data = data.astype(np.float32) data = data.astype(np.float32)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment