14d34e59b208800c355c68118ee7f41947364905,ch07/lib/dqn_model.py,NoisyLinear,sample_noise,#NoisyLinear#,32

Before Change

    def sample_noise(self):
        self.epsilon_weight = torch.randn(self.out_features, self.in_features)
        self.epsilon_bias = torch.randn(self.out_features)

    def remove_noise(self):
        self.epsilon_weight = torch.zeros(self.out_features, self.in_features)
        self.epsilon_bias = torch.zeros(self.out_features)

After Change

                        self.bias + self.sigma_bias * Variable(self.epsilon_bias))

    def sample_noise(self):
        torch.randn(self.epsilon_weight.size(), out=self.epsilon_weight)
        torch.randn(self.epsilon_bias.size(), out=self.epsilon_bias)


class DQN(nn.Module):

Instances