More work on tring to make torch work

logic/models/train/torch/utils.go

@@ -107,6 +107,7 @@ func NewFlatten() *Flatten {
 
 // The flatten layer does not to move anything to the device
 func (b *Flatten) ExtractFromVarstore(vs *my_nn.VarStore) {}
+func (b *Flatten) Debug() {}
 
 // Forward method
 func (b *Flatten) Forward(x *torch.Tensor) *torch.Tensor {
@@ -133,6 +134,7 @@ func NewSigmoid() *Sigmoid {
 
 // The sigmoid layer does not need to move anything to another device
 func (b *Sigmoid) ExtractFromVarstore(vs *my_nn.VarStore) {}
+func (b *Sigmoid) Debug() {}
 
 func (b *Sigmoid) Forward(x *torch.Tensor) *torch.Tensor {
 	out, err := x.Sigmoid(false)