Export_model.py File Reference

Namespaces
namespace	Export_model

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