andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chores

Browse Source 
(cleanup)
This commit is contained in:
sugarme 2020-08-05 20:31:01 +10:00
parent 17b9e6c46f
commit f2fd373edf
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nn/init.go
View File

@ -163,7 +163,6 @@ func (k kaimingUniformInit) InitTensor(dims []int64, device gotch.Device) (retVa
}
bound := math.Sqrt(1.0 / float64(fanIn))
log.Println(fanIn)
kind := gotch.Float
retVal = ts.MustZeros(dims, kind, device)
retVal.Uniform_(-bound, bound)