Add eps to normalization (#797)
This commit is contained in:
@@ -17,7 +17,7 @@ except ImportError:
|
|||||||
def normalize(x, stats):
|
def normalize(x, stats):
|
||||||
if stats is None:
|
if stats is None:
|
||||||
return x
|
return x
|
||||||
return (x - stats.mean) / stats.std
|
return (x - stats.mean) / (stats.std + 1e-8)
|
||||||
|
|
||||||
|
|
||||||
def denormalize(x, stats):
|
def denormalize(x, stats):
|
||||||
|
Reference in New Issue
Block a user