add weight

Author: andrewk1

bandits/algorithms/neural_bandit_model.py

@@ -20,7 +20,7 @@ def __init__(self, optimizer, hparams, name):
     def build_layer(self, input_dim, output_dim, last=False):
         """Builds a fc layer with num_inputs and num_outputs"""
         layer = [nn.Linear(input_dim, output_dim)]
-        torch.nn.init.uniform_(layer[0], -0.3, 0.3)
+        torch.nn.init.uniform_(layer[0].weight, -0.3, 0.3)
         if not last: layer += [nn.ReLU()]
         # TODO: Maybe add layer_norm
         if self.hparams.keep_prob < 1.: