diff --git a/README.md b/README.md
new file mode 100644
index 0000000..1470cea
--- /dev/null
+++ b/README.md
@@ -0,0 +1,5 @@
+
+
+# Resources
+
+Import ONXX into https://crates.io/crates/burn
\ No newline at end of file
diff --git a/train.ipynb b/train.ipynb
index 2249ea7..c77e1b3 100644
--- a/train.ipynb
+++ b/train.ipynb
@@ -2,16 +2,16 @@
"cells": [
{
"cell_type": "code",
- "execution_count": 1,
+ "execution_count": 14,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
- "'Using device `cuda:0`'"
+ "'Using device cuda:0'"
]
},
- "execution_count": 1,
+ "execution_count": 14,
"metadata": {},
"output_type": "execute_result"
}
@@ -24,22 +24,22 @@
"CACHE_PATH = Path('/mnt/wsl/PHYSICALDRIVE1/data/cache2')\n",
"CACHE_PATH.mkdir(exist_ok=True, parents=True)\n",
"BINS = 32\n",
- "NUM_EPOCHS = 10\n",
- "BATCH_SIZE = 32\n",
+ "NUM_EPOCHS = 20\n",
+ "BATCH_SIZE = 64\n",
"LEARNING_RATE = 0.005\n",
- "SCHEDULER_GAMMA = 0.2\n",
+ "SCHEDULER_GAMMA = 0.7\n",
"EDIT_COUNT = 25\n",
"LOSS_DAMPING = 2\n",
"MODELS_PATH = Path('models')\n",
"MODELS_PATH.mkdir(exist_ok=True, parents=True)\n",
"\n",
"device = torch.device(\"cuda:0\") if torch.cuda.is_available() else torch.device(\"cpu\")\n",
- "f'Using device `{device}`'"
+ "f'Using device {device}'"
]
},
{
"cell_type": "code",
- "execution_count": 2,
+ "execution_count": 15,
"metadata": {},
"outputs": [
{
@@ -48,7 +48,7 @@
"'Loaded 561982 training images and 62443 test images'"
]
},
- "execution_count": 2,
+ "execution_count": 15,
"metadata": {},
"output_type": "execute_result"
}
@@ -71,7 +71,7 @@
},
{
"cell_type": "code",
- "execution_count": 3,
+ "execution_count": 16,
"metadata": {},
"outputs": [],
"source": [
@@ -148,7 +148,7 @@
},
{
"cell_type": "code",
- "execution_count": 4,
+ "execution_count": 17,
"metadata": {},
"outputs": [],
"source": [
@@ -180,26 +180,37 @@
" ax.set_zlim([0, 31])\n",
"\n",
" ax.set_title(f'Tensor {i}')\n",
- " return fig\n",
- "\n",
- "\n",
- "# plot_histograms(original_histogram, edited_histogram, edited_histogram)"
+ " return fig"
]
},
{
"cell_type": "code",
- "execution_count": 5,
+ "execution_count": 18,
"metadata": {},
"outputs": [
{
"data": {
"application/vnd.jupyter.widget-view+json": {
- "model_id": "b1a1b0d47c494d52af3f9a6a8eda7127",
+ "model_id": "065d54e849f54398904e9de25718d85a",
"version_major": 2,
"version_minor": 0
},
"text/plain": [
- "Epoch 0: 0%| | 0/17562 [00:00