0706-2044

diffusion.ipynb

@@ -255,14 +255,14 @@
     "    device = 'cuda' if torch.cuda.is_available() else 'cpu'\n",
     "    # repeat = 2\n",
     "\n",
-    "    dim = 2\n",
-    "    # dim = 3\n",
+    "    # dim = 2\n",
+    "    dim = 3\n",
     "    stride = (2,2) if dim == 2 else (2,2,2)\n",
     "    num_image = 2000#32000#20000#15000#7000#25600#3000#10000#1000#10000#5000#2560#800#2560\n",
-    "    batch_size = 2*64#2#50#20#2#100 # 10\n",
+    "    batch_size = 2#2#50#20#2#100 # 10\n",
     "    n_epoch = 10#50#20#20#2#5#25 # 120\n",
     "    HII_DIM = 64\n",
-    "    num_redshift = 64#512#256#256#64#512#128\n",
+    "    num_redshift = 128#64#512#256#256#64#512#128\n",
     "    channel = 1\n",
     "    img_shape = (channel, HII_DIM, num_redshift) if dim == 2 else (channel, HII_DIM, HII_DIM, num_redshift)\n",
     "\n",
@@ -565,15 +565,14 @@
      "output_type": "stream",
      "text": [
       "Number of parameters for nn_model: 111048705\n",
-      "---------------- num_image = 512 -----------------\n",
-      "run_name = 0706-2033\n",
+      "----------------- num_image = 20 -----------------\n",
+      "run_name = 0706-2044\n",
       "Launching training on one GPU.\n",
       "dataset content: <KeysViewHDF5 ['brightness_temp', 'density', 'kwargs', 'params', 'redshifts_distances', 'seeds', 'xH_box']>\n",
       "51200 images can be loaded\n",
       "field.shape = (64, 64, 514)\n",
       "params keys = [b'ION_Tvir_MIN', b'HII_EFF_FACTOR']\n",
-      "loading 512 images randomly\n",
-      "images loaded: (512, 1, 64, 64)\n"
+      "loading 20 images randomly\n"
      ]
     },
     {
@@ -587,20 +586,21 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "params loaded: (512, 2)\n",
-      "images rescaled to [-1.0, 1.116640567779541]\n",
-      "params rescaled to [0.00024580534048324054, 0.9967556649676916]\n"
+      "images loaded: (20, 1, 64, 128)\n",
+      "params loaded: (20, 2)\n",
+      "images rescaled to [-1.0, 1.0278661251068115]\n",
+      "params rescaled to [0.005739769005289105, 0.972333144312969]\n"
      ]
     },
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "0528cc7f8f7b4013a38dc1f5f5ecc99b",
+       "model_id": "432face5514f46dbab6a0bd6752ee9dd",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -609,12 +609,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "03c91703f35749bdb710e54196d0cd8e",
+       "model_id": "43bc633a08524f1ea75546311244f8aa",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -623,12 +623,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "f55a5c7a47074aa49326453b7b3d9a2b",
+       "model_id": "884b273aa5d9468ba1facd835f4de2df",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -637,12 +637,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "4f41ba4f484b4548abc4a8694fd51fcf",
+       "model_id": "7eabba7c6741415ea5e6fd097e33dbb2",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -651,12 +651,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "5148cb8ca78b4787a533c420adffa654",
+       "model_id": "8cfc863995a74070b23bfcd73d314cca",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -665,12 +665,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "ee1acc3b3dff479c973a66747f5a674d",
+       "model_id": "7c53f6b374644535aa8e2bce7e961d80",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -679,12 +679,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "beea681d6de64561855a3ae95318fe7a",
+       "model_id": "2494d9b658f7493088b64eab1006f2c4",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -693,12 +693,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "b627c0db8ba142eb8a11db3ba1d2cccd",
+       "model_id": "070d1da1b37043f8aac851d3bf5c22f3",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -707,12 +707,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "ec52b0207eb049f6aff31f5847283d5b",
+       "model_id": "063fc075914d4e7bb5ffaf9f0ac14285",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -721,12 +721,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "e959387839db4d3b8a601723e19f7f35",
+       "model_id": "01c9c1d56a394f3caf0c6fec1b072bcd",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/10 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -734,7 +734,7 @@
     }
    ],
    "source": [
-    "num_image_list = [512]#[20]#[200]#[1600,3200,6400,12800,25600]\n",
+    "num_image_list = [20]#[200]#[1600,3200,6400,12800,25600]\n",
     "if __name__ == \"__main__\":\n",
     "    # torch.multiprocessing.set_start_method(\"spawn\")\n",
     "    # args = (config, nn_model, ddpm, optimizer, dataloader, lr_scheduler)\n",