diff --git a/unsuper/conv1_output.png b/unsuper/conv1_output.png index 90c2999..37a8ff0 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 120638b..8c91dad 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 3f1809f..b96d0e8 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 7b9d69e..bddfe9c 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 dc15694..6fa8ecf 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 032b31b..ebdebc9 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 00316ef..d5cab67 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 516ca3f..2f3e4c6 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 8a2c5db..bedab05 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 b75b229..5a6af5c 100644 --- a/unsuper/minist.py +++ b/unsuper/minist.py @@ -18,8 +18,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # Hyper-parameters num_epochs = 1 -batch_size = 256 -learning_rate = 0.001 +batch_size = 128 +learning_rate = 0.2 transform = transforms.Compose([transforms.ToTensor()])