0523-0028

diffusion.ipynb

@@ -225,7 +225,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -243,7 +243,7 @@
     "    # dim = 2\n",
     "    dim = 2\n",
     "    stride = (2,2) if dim == 2 else (2,2,4)\n",
-    "    num_image = 40#2400\n",
+    "    num_image = 240#0\n",
     "    HII_DIM = 64\n",
     "    num_redshift = 512#256#256#64#512#128\n",
     "    channel = 1\n",
@@ -291,7 +291,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [
     {
@@ -444,7 +444,7 @@
     "                            repo_id = self.repo_id,\n",
     "                            folder_path = \".\",#config.output_dir,\n",
     "                            commit_message = f\"{self.config.date}\",\n",
-    "                            ignore_patterns = [\"step_*\", \"epoch_*\", \"*.npy\"],\n",
+    "                            ignore_patterns = [\"step_*\", \"epoch_*\", \"*.npy\", \"__pycache__\"],\n",
     "                            )\n",
     "                    if self.config.save_model:\n",
     "                        model_state = {\n",
@@ -502,13 +502,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "d488e670a37b408399687972aa7fef8a",
+       "model_id": "16b1b0f49854415fbcc0b742cc453b7d",
        "version_major": 2,
        "version_minor": 0
       },
@@ -528,8 +528,8 @@
       "51200 images can be loaded\n",
       "field.shape = (64, 64, 514)\n",
       "params keys = [b'ION_Tvir_MIN', b'HII_EFF_FACTOR']\n",
-      "loading 40 images randomly\n",
-      "images loaded: (40, 1, 64, 512)\n"
+      "loading 240 images randomly\n",
+      "images loaded: (240, 1, 64, 512)\n"
      ]
     },
     {
@@ -543,20 +543,20 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "params loaded: (40, 2)\n",
-      "images rescaled to [-1.0, 1.1514630317687988]\n",
-      "params rescaled to [0.0, 0.9958124549229699]\n"
+      "params loaded: (240, 2)\n",
+      "images rescaled to [-1.0, 1.1050198078155518]\n",
+      "params rescaled to [0.0, 0.9977353610767956]\n"
      ]
     },
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "09696eb258f94136b32dec3290034ce2",
+       "model_id": "5d938426444b466e8a1887e9f461fee4",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -565,12 +565,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "77a8a7ad49d84bdcbfc522069118b222",
+       "model_id": "eb012055296f4d0fa03b6ede4160f2f9",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -579,12 +579,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "28a00d5f8fdc472eae5b252888410421",
+       "model_id": "cb486b58cb0a43b9a5dca19d78617e8f",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -593,12 +593,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "f145f20dab944419a730632a9d5081fc",
+       "model_id": "afd120fad0a54479afd403350e1c47f1",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -607,12 +607,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "de801afbf6aa428b9d21bf6a76a28ee8",
+       "model_id": "6878fbbd5d36411caa80f23124a070fa",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -621,12 +621,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "1e2c7ec7c4df438c93f21b420e93aaf3",
+       "model_id": "8fcfc88ca22c4c7cbb6e847bd8c1a226",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -635,12 +635,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "e963fcaf27ae401e926053f9f9c146f4",
+       "model_id": "71906ea928b0444888e43d8e1b6b1784",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -649,12 +649,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "d6fb1a76c93b45e59fa7d253ef344bda",
+       "model_id": "0d8745fe8157402db413d318a2e50867",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -663,12 +663,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "f0d88179dc7d42a793a820f4d1a08da3",
+       "model_id": "5adde60a2f974c4785a5e1e14c0b0214",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -677,12 +677,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "c8caffa8edc6457e97c2d315252a5adf",
+       "model_id": "cda697db8ae2422bb27bc4e39edd893d",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
+       "  0%|          | 0/24 [00:00<?, ?it/s]"
       ]
      },
      "metadata": {},
@@ -698,7 +698,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {