Memory leaks about gotch HOT 4 CLOSED

package main

import (
	"log"

	"github.com/sugarme/gotch"
	"github.com/sugarme/gotch/ts"
	"github.com/sugarme/gotch/vision"
)

type Model struct {
	cmodule  *ts.CModule
	imageNet *vision.ImageNet
}

func NewModel(modelPath string) (*Model, error) {
	model, err := ts.ModuleLoadOnDevice(modelPath, gotch.CPU)
	if err != nil {
		return nil, err
	}

	imageNet := vision.NewImageNet()

	return &Model{
		cmodule:  model,
		imageNet: imageNet,
	}, nil
}

func (m *Model) Predict(imageFile string) ([]float32, error) {
	imageNet := vision.NewImageNet()
	image, err := imageNet.LoadImageAndResize224(imageFile)
	if err != nil {
		log.Fatal(err)
	}
	input := image.MustUnsqueeze(0, true)
	raw_output := input.ApplyCModule(m.cmodule)
	output := raw_output.MustSoftmax(-1, gotch.Float, true)

	probs := output.Vals().([]float32)
	output.MustDrop()

	return probs, nil
}

func main() {
	modelPath := "./model.pt"
	imageFile := "./image.jpg"

	model, err := NewModel(modelPath)
	if err != nil {
		panic(err)
	}

	n := 1000
	for i := 0; i < n; i++ {
		prob, err := model.Predict(imageFile)
		if err != nil {
			panic(err)
		}

		if i%100 == 0 && i > 0 {
			log.Printf("Done... %2d: %v\n", i, prob)
		}
	}
}