fyp/logic/models/train/torch/nn/utils.go

19 lines
233 B
Go
Raw Normal View History

2024-04-22 00:09:07 +01:00
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)
2024-04-23 00:14:35 +01:00
Debug()
2024-04-22 00:09:07 +01:00
}