소스 검색

tensor loading iface

Jesse Gross 3 달 전
부모
커밋
3c95c21ddf
1개의 변경된 파일10개의 추가작업 그리고 0개의 파일을 삭제
  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
 }