Commit 600d8dc3 authored by okiss's avatar okiss
Browse files

Update CNN.py

parent c5146bb2
......@@ -47,6 +47,7 @@ def run(trainX, trainY):
# Newly added lines below
save_logs(hist, pytorch=True)
save_model_param(pytorch=True)
plot_loss_torch(loss)
def train(trainloader, net, optimizer, criterion, epoch=50):
loss=[]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment