andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-04 22:51:20 +01:00
parent edeac12e37
commit ea7931fc57
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
1 changed files with 11 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

65
src/train.ipynb
View file

@ -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",