models Namespace Reference

Namespaces
	meta

Functions
def	parse_model_args (model_name, parser, add_help=False)
def	batchnorm_to_float (module)
def	init_bn (module)
def	get_model (model_name, model_config, to_cuda, uniform_initialize_bn_weight=False, forward_is_infer=False)
def	get_model_config (model_name, args)

Function Documentation

parse_model_args()

def models.parse_model_args	(		model_name,
			parser,
			add_help = False
	)

Here is the caller graph for this function:

batchnorm_to_float()

def models.batchnorm_to_float

(

module

)

Converts batch norm to FP32

init_bn()

def models.init_bn

(

module

)

Here is the caller graph for this function:

get_model()

def models.get_model	(		model_name,
			model_config,
			to_cuda,
			uniform_initialize_bn_weight = False,
			forward_is_infer = False
	)

Code chooses a model based on name

Here is the call graph for this function:

Here is the caller graph for this function:

get_model_config()

def models.get_model_config	(		model_name,
			args
	)

Code chooses a model based on name

Here is the caller graph for this function: