More work on tring to make torch work

2024-04-23 00:14:35 +01:00
parent 703fea46f2
commit a4a9ade71f
7 changed files with 109 additions and 43 deletions
--- a/logic/models/train/torch/nn/utils.go
+++ b/logic/models/train/torch/nn/utils.go
@@ -14,4 +14,5 @@ type MyLayer interface {
    torch.ModuleT

    ExtractFromVarstore(vs *VarStore)
+    Debug()
 }