Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
iperov
GitHub Repository: iperov/deepfacelab
Path: blob/master/core/randomex.py
628 views
1
import numpy as np
2
3
def random_normal( size=(1,), trunc_val = 2.5, rnd_state=None ):
4
if rnd_state is None:
5
rnd_state = np.random
6
len = np.array(size).prod()
7
result = np.empty ( (len,) , dtype=np.float32)
8
9
for i in range (len):
10
while True:
11
x = rnd_state.normal()
12
if x >= -trunc_val and x <= trunc_val:
13
break
14
result[i] = (x / trunc_val)
15
16
return result.reshape ( size )
17