def __init__(self, state_model, object_model, heatmap_model, args, map_dim = 10):
super(MultiModel, self).__init__()
self.state_model = state_model
self.object_model = object_model
self.heatmap_model = heatmap_model
self.simple_conv = models.SimpleConv(2).cuda()
self.rbf = Variable( utils.meta_rbf(map_dim).cuda() )
self.map_dim = map_dim
self.batch_size = args.batch_size
self.rbf_batch = self.rbf.repeat(self.batch_size,1,1,1)
def __init__(self,
state_model,
object_model,
heatmap_model,
args,
map_dim=10):
super(MultiNoRBF, self).__init__()
self.state_model = state_model
self.object_model = object_model
self.heatmap_model = heatmap_model
self.simple_conv = cudit(models.SimpleConv(3))
self.rbf = cudit(Variable(utils.meta_rbf(map_dim)))
self.positions = cudit(Variable(self.__init_positions(map_dim)))
self.map_dim = map_dim
self.batch_size = args.batch_size
self.rbf_batch = self.rbf.repeat(self.batch_size, 1, 1, 1)
self.positions_batch = self.positions.repeat(self.batch_size, 1, 1, 1)
