gotch/example/mnist/main.go

package main

import (
	"flag"
)

var model string

func init() {
	flag.StringVar(&model, "model", "linear", "specify a model to run")

}

func main() {

	flag.Parse()

	switch model {
	case "linear":
		runLinear()
	case "nn":
		runNN()
	case "cnn":
		runCNN2()
	default:
		panic("No specified model to run")
	}

}