diff --git "a/diffusion.ipynb" "b/diffusion.ipynb" --- "a/diffusion.ipynb" +++ "b/diffusion.ipynb" @@ -247,7 +247,7 @@ " stride = (2,2) if dim == 2 else (2,2,4)\n", " num_image = 800#2560\n", " batch_size = 10#20#2#100 # 10\n", - " n_epoch = 20#2#5#25 # 120\n", + " n_epoch = 40#20#2#5#25 # 120\n", " HII_DIM = 64\n", " num_redshift = 64#512#256#256#64#512#128\n", " channel = 1\n", @@ -500,7 +500,7 @@ { "data": { "application/vnd.jupyter.widget-view+json": { - "model_id": "ce26407b5c21438a8328585b8715a67b", + "model_id": "a89cee1d5c8644998aa65ddc6e7e7302", "version_major": 2, "version_minor": 0 }, @@ -526,8 +526,10 @@ "output_type": "stream", "text": [ "-------------------- round 0 ---------------------\n", + "resumed nn_model from ./outputs/model_state.pth\n", "Number of parameters for nn_model: 111048705\n", - "run_name = 0528-1406\n", + "resumed ema_model from ./outputs/model_state.pth\n", + "run_name = 0528-1423\n", "Launching training on one GPU.\n", "dataset content: \n", "1000 images can be loaded\n", @@ -549,14 +551,182 @@ "output_type": "stream", "text": [ "params loaded: (800, 2)\n", - "images rescaled to [-1.0, 0.07598578929901123]\n", + "images rescaled to [-1.0, 0.06256699562072754]\n", "params rescaled to [0.20000000000000018, 0.5055875000000001]\n" ] }, { "data": { "application/vnd.jupyter.widget-view+json": { - "model_id": "42de52767a6d4ad59dd5183616732aa4", + "model_id": "8830fc42adc746a7ae29e1a788f119fa", + "version_major": 2, + "version_minor": 0 + }, + "text/plain": [ + " 0%| | 0/80 [00:00" ] @@ -977,7 +1261,7 @@ } ], "source": [ - "plot(\"outputs/0527-2152.npy\")\n", + "plot(\"outputs/0528-1416.npy\")\n", "# plot(\"outputs/0520-2323.npy\")" ] },