From ea7931fc57681a3919acfa17a1de77ab4c7d0e92 Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Tue, 4 Jun 2024 22:51:20 +0100
Subject: [PATCH] Update

---
 src/train.ipynb | 65 +++++++++----------------------------------------
 1 file changed, 11 insertions(+), 54 deletions(-)

diff --git a/src/train.ipynb b/src/train.ipynb
index 6e32210..7397bbf 100644
--- a/src/train.ipynb
+++ b/src/train.ipynb
@@ -2,22 +2,11 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {
     "metadata": {}
    },
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'Using device cuda:0'"
-      ]
-     },
-     "execution_count": 1,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "import torch\n",
     "import logging\n",
@@ -29,7 +18,9 @@
     "    format=\"%(asctime)s - %(levelname)s - %(message)s\",\n",
     "    handlers=[\n",
     "        logging.StreamHandler(),\n",
-    "        logging.FileHandler(f\"train-{datetime.now().date()}.log\"),\n",
+    "        logging.FileHandler(\n",
+    "            f\"train-{datetime.now().isoformat(timespec='minutes')}.log\"\n",
+    "        ),\n",
     "    ],\n",
     ")\n",
     "\n",
@@ -40,7 +31,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -74,7 +65,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -106,7 +97,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -292,7 +283,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -315,43 +306,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "2024-06-04 22:42:58,542 - INFO - Starting run_6 with hparams {\n",
-      "  \"batch_size\": 32,\n",
-      "  \"bin_count\": 24,\n",
-      "  \"clip_gradients\": true,\n",
-      "  \"edit_count\": 16,\n",
-      "  \"learning_rate\": 0.00043772328325342977,\n",
-      "  \"loss\": \"kl\",\n",
-      "  \"model_type\": \"Residual\",\n",
-      "  \"num_epochs\": 10,\n",
-      "  \"scheduler_gamma\": 0.7839153264198727\n",
-      "}\n",
-      "2024-06-04 22:42:58,546 - INFO - Loaded 800 original images\n",
-      "2024-06-04 22:42:58,548 - INFO - Loaded 200 original images\n",
-      "2024-06-04 22:43:28,238 - INFO - Epoch 0 train loss: 1092.8763027191162\n",
-      "2024-06-04 22:43:32,819 - INFO - Epoch 0 test loss: 378.1922540664673\n",
-      "2024-06-04 22:43:58,916 - INFO - Epoch 1 train loss: 833.7781882286072\n",
-      "2024-06-04 22:44:03,764 - INFO - Epoch 1 test loss: 415.40036368370056\n",
-      "2024-06-04 22:44:29,857 - INFO - Epoch 2 train loss: 767.9900119304657\n",
-      "2024-06-04 22:44:34,696 - INFO - Epoch 2 test loss: 297.7264175415039\n",
-      "2024-06-04 22:45:00,728 - INFO - Epoch 3 train loss: 726.7450095415115\n",
-      "2024-06-04 22:45:05,531 - INFO - Epoch 3 test loss: 265.4522042274475\n",
-      "2024-06-04 22:45:31,520 - INFO - Epoch 4 train loss: 700.53100669384\n",
-      "2024-06-04 22:45:36,503 - INFO - Epoch 4 test loss: 285.5881419181824\n",
-      "2024-06-04 22:46:02,685 - INFO - Epoch 5 train loss: 675.3199667930603\n",
-      "2024-06-04 22:46:07,717 - INFO - Epoch 5 test loss: 246.0367443561554\n",
-      "2024-06-04 22:46:33,992 - INFO - Epoch 6 train loss: 656.3262705802917\n",
-      "2024-06-04 22:46:38,921 - INFO - Epoch 6 test loss: 242.63626527786255\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "from random import choice\n",
     "from itertools import count\n",