Refine train.

2024-09-01 18:15:07 +08:00 · 2024-09-01 18:15:07 +08:00 · 4ad4c01403
parent c395fa7baa
commit 4ad4c01403
10 changed files with 2 additions and 2 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,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()])