Update high parameter of minist.

2024-09-02 18:06:47 +08:00 · 2024-09-02 18:06:47 +08:00 · a052fceaeb
parent e48e724ca9
commit a052fceaeb
11 changed files with 1 additions and 1 deletions
--- a/unsuper/conv1_output.png
+++ b/unsuper/conv1_output.png
--- a/unsuper/conv1_weight.png
+++ b/unsuper/conv1_weight.png
--- a/unsuper/conv1_weight_grad.png
+++ b/unsuper/conv1_weight_grad.png
--- a/unsuper/conv2_output.png
+++ b/unsuper/conv2_output.png
--- a/unsuper/conv2_weight.png
+++ b/unsuper/conv2_weight.png
--- a/unsuper/conv2_weight_grad.png
+++ b/unsuper/conv2_weight_grad.png
--- a/unsuper/fc_output.png
+++ b/unsuper/fc_output.png
--- a/unsuper/fc_weight.png
+++ b/unsuper/fc_weight.png
--- a/unsuper/fc_weight_grad.png
+++ b/unsuper/fc_weight_grad.png
--- 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()])
--- a/unsuper/pool_output.png
+++ b/unsuper/pool_output.png