Export_model Namespace Reference

Variables
	MVDepthNet = depthNet()
	weights = torch.load('depthNet_ep48_model_best.pth.tar')
	pretrained_dict = weights['model_state_dict']
	model = torch.quantization.convert(MVDepthNet)
	x_ = torch.rand(1, 3, 352, 480)
	cv_ = torch.rand(1, 64, 352, 480)
	scripted_model = torch.jit.trace(MVDepthNet, [x_, cv_])
	opt_model = mobile_optimizer.optimize_for_mobile(scripted_model)

Variable Documentation

cv_

Export_model.cv_ = torch.rand(1, 64, 352, 480)

model

Export_model.model = torch.quantization.convert(MVDepthNet)

MVDepthNet

Export_model.MVDepthNet = depthNet()

opt_model

Export_model.opt_model = mobile_optimizer.optimize_for_mobile(scripted_model)

pretrained_dict

Export_model.pretrained_dict = weights['model_state_dict']

scripted_model

Export_model.scripted_model = torch.jit.trace(MVDepthNet, [x_, cv_])

weights

Export_model.weights = torch.load('depthNet_ep48_model_best.pth.tar')

x_

Export_model.x_ = torch.rand(1, 3, 352, 480)