Browse Source

tensor loading iface

Jesse Gross 3 months ago
parent
commit
a34960a516
1 changed files with 10 additions and 0 deletions
  1. 10 0
      model/model.go

+ 10 - 0
model/model.go

@@ -152,6 +152,12 @@ func New(s string) (Model, error) {
 }
 
 func populateFields(b ml.Backend, v reflect.Value, tags ...Tag) reflect.Value {
+	var iface bool
+	if v.Kind() == reflect.Interface {
+		iface = true
+		v = v.Elem()
+	}
+
 	t := v.Type()
 	if t.Kind() == reflect.Pointer {
 		t, v = t.Elem(), v.Elem()
@@ -230,6 +236,10 @@ func populateFields(b ml.Backend, v reflect.Value, tags ...Tag) reflect.Value {
 		}
 	}
 
+	if iface {
+		return v.Addr()
+	}
+
 	return v
 }