def sample_model_seed(self):
    if (not self.mixer):
        self.model_seed = {'fc_head_z': torch.rand(self.fc_head.config['n_gen'], particles, self.z_dim).to(Config.DEVICE)}
    else:
        self.model_seed = torch.rand(particles, self.s_dim)
def sample_model_seed(self):
    if (not self.mixer):
        self.model_seed = {'fc_head_z': torch.rand(self.fc_head.config['n_gen'], particles, self.z_dim).to(Config.DEVICE)}
    else:
        self.model_seed = torch.rand(particles, self.s_dim)