diff --git a/unsuper/conv1_output.png b/unsuper/conv1_output.png index e0b1f3c..c21c3de 100644 Binary files a/unsuper/conv1_output.png and b/unsuper/conv1_output.png differ diff --git a/unsuper/conv1_weight.png b/unsuper/conv1_weight.png index ec2d4e4..e3e5bb5 100644 Binary files a/unsuper/conv1_weight.png and b/unsuper/conv1_weight.png differ diff --git a/unsuper/conv1_weight_grad.png b/unsuper/conv1_weight_grad.png index 80168ee..61a57a7 100644 Binary files a/unsuper/conv1_weight_grad.png and b/unsuper/conv1_weight_grad.png differ diff --git a/unsuper/conv2_output.png b/unsuper/conv2_output.png index 550436f..a4880cd 100644 Binary files a/unsuper/conv2_output.png and b/unsuper/conv2_output.png differ diff --git a/unsuper/conv2_weight.png b/unsuper/conv2_weight.png index 7bdc554..a62244e 100644 Binary files a/unsuper/conv2_weight.png and b/unsuper/conv2_weight.png differ diff --git a/unsuper/conv2_weight_grad.png b/unsuper/conv2_weight_grad.png index 1fdc005..14dbcec 100644 Binary files a/unsuper/conv2_weight_grad.png and b/unsuper/conv2_weight_grad.png differ diff --git a/unsuper/fc_output.png b/unsuper/fc_output.png index 5c1849e..b5cf48c 100644 Binary files a/unsuper/fc_output.png and b/unsuper/fc_output.png differ diff --git a/unsuper/fc_weight.png b/unsuper/fc_weight.png index d2abf1c..2336024 100644 Binary files a/unsuper/fc_weight.png and b/unsuper/fc_weight.png differ diff --git a/unsuper/fc_weight_grad.png b/unsuper/fc_weight_grad.png index 57b5b6d..57eef8d 100644 Binary files a/unsuper/fc_weight_grad.png and b/unsuper/fc_weight_grad.png differ diff --git a/unsuper/minist.py b/unsuper/minist.py index 3cff4d3..3a98061 100644 --- a/unsuper/minist.py +++ b/unsuper/minist.py @@ -18,7 +18,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # Hyper-parameters num_epochs = 1 -batch_size = 128 +batch_size = 64 learning_rate = 0.2 transform = transforms.Compose([transforms.ToTensor()]) diff --git a/unsuper/pool_output.png b/unsuper/pool_output.png index c4eba86..bb629c9 100644 Binary files a/unsuper/pool_output.png and b/unsuper/pool_output.png differ