More work done on torch
This commit is contained in:
17
logic/models/train/torch/nn/utils.go
Normal file
17
logic/models/train/torch/nn/utils.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package my_nn
|
||||
|
||||
import (
|
||||
torch "git.andr3h3nriqu3s.com/andr3/gotch/ts"
|
||||
)
|
||||
|
||||
func or_panic(err error) {
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
type MyLayer interface {
|
||||
torch.ModuleT
|
||||
|
||||
ExtractFromVarstore(vs *VarStore)
|
||||
}
|
||||
Reference in New Issue
Block a user