parser

Documentation

Index

• func CheckModel(cfgName string, paramsName string) error
• type Block
  • func (b *Block) OutputBlock()
• type Layer
• type LayerCode
• type Parser
  • func (p *Parser) GenBlock() (err error)
  • func (p *Parser) ImportFile(cfgName string, modelName string) error
  • func (p *Parser) OutputBlocks()
  • func (p *Parser) ParseCFG() (size int64, err error)
• type UnitCode
• type ValueType

Functions

func CheckModel

func CheckModel(cfgName string, paramsName string) error

Types

type Block

type Block struct {
	// contains filtered or unexported fields
}

func (*Block) OutputBlock

func (b *Block) OutputBlock()

type Layer

type Layer struct {
	// contains filtered or unexported fields
}

type LayerCode

type LayerCode byte

const (
	LC_NET LayerCode = iota
	LC_CONV
	LC_ACT
	LC_FC
	LC_RES
	LC_MAXPOOL
)

type Parser

type Parser struct {
	// contains filtered or unexported fields
}

func (*Parser) GenBlock

func (p *Parser) GenBlock() (err error)

GenBlock load each block

func (*Parser) ImportFile

func (p *Parser) ImportFile(cfgName string, modelName string) error

func (*Parser) OutputBlocks

func (p *Parser) OutputBlocks()

func (*Parser) ParseCFG

func (p *Parser) ParseCFG() (size int64, err error)

type UnitCode

type UnitCode byte

const (
	UC_BATCH UnitCode = iota
	UC_SUBDIVISIONS
	UC_HEIGHT
	UC_WIDTH
	UC_CHANNELS
	UC_FILTERS
	UC_SIZE
	UC_STRIDE
	UC_PAD
	UC_ACTIVATION
	UC_OUTPUT
	UC_FROM
)

type ValueType

type ValueType byte

const (
	STR ValueType = iota
	INT
	ERR
)