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