User submitted

2020-10-24 17:00:42 +08:00 · 2020-10-24 17:00:42 +08:00 · 4624d3590e
parent 9b10af2116
commit 4624d3590e
1 changed files with 1 additions and 1 deletions
--- a/step2/step2.py
+++ b/step2/step2.py
@ -46,7 +46,7 @@ def dataset_loader(data_path, batch_size=32, repeat_size=1, num_parallel_workers
    resize_height, resize_width = 32, 32
    rescale = 1.0 / 255.0
    #!
-    shift = 0.005
+    shift = 0
    rescale_nml = 1 / 0.3081
    shift_nml = -1 * 0.1307 / 0.3081
    ########## end ##########