More work on the torch version

2024-04-23 11:54:30 +01:00
parent a4a9ade71f
commit 527b57a111
4 changed files with 109 additions and 102 deletions
--- a/logic/models/train/torch/torch.go
+++ b/logic/models/train/torch/torch.go
@@ -27,6 +27,7 @@ func (n *ContainerModel) ForwardT(x *torch.Tensor, train bool) *torch.Tensor {
 	}

 	if len(n.Layers) == 1 {
+        log.Info("here")
 		return n.Layers[0].ForwardT(x, train)
 	}