change

whitecatdry · Oct 24, 2021 · 3267013 · 3267013
1 parent e19eb7a
commit 3267013
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/CNN_multichannel.py b/CNN_multichannel.py
@@ -152,8 +152,9 @@ def forward(self, x):
         return logit
 
 batch_size = 2
-learning_rate = 1e-6
+learning_rate = 8e-5
 num_epoches = 40000
+valid_loss_min = 2
 
 model = Cnn1d(7)
 
@@ -168,9 +169,8 @@ def forward(self, x):
 
 loss_curve = []
 tr_acc = []
-valid_loss_min = 2
 # train
-epoch = 1
+epoch = 0
 while epoch < num_epoches:
     train_acc = 0
     for i in range(len(X_tensor)):
@@ -207,6 +207,8 @@ def forward(self, x):
 
         # print('epoch: {}, loss: {:.4}'.format(epoch, print_loss), 'step: ', i + 1)
 
+    epoch += 1
+
     # calculate accuracy
     acc = train_acc / len(X_tensor)
     tr_acc.append(acc)
@@ -215,7 +217,7 @@ def forward(self, x):
 
     if epoch % 10 == 0:
         print('epoch: {}, loss: {:.4}, acc: {:.4}'.format(epoch, print_loss, acc))
-        print(out, '->', pred, ':', label - 1, loss)
+        # print(out, '->', pred, ':', label - 1, loss)
 
         # create checkpoint variable and add important data
         checkpoint = {
@@ -231,7 +233,6 @@ def forward(self, x):
             torch.save(model.state_dict(), 'model_best')
             valid_loss_min = print_loss
 
-    epoch += 1
 
 plt.plot(loss_curve)
 fig2 = plt.gcf()