0709-1355

diffusion.ipynb

@@ -399,6 +399,7 @@
     "        ## training loop ##\n",
     "        ###################\n",
     "        # plot_unet = True\n",
+    "        \n",
     "        self.load()\n",
     "        self.accelerator = Accelerator(\n",
     "            mixed_precision=self.config.mixed_precision,\n",
@@ -406,6 +407,7 @@
     "            log_with=\"tensorboard\",\n",
     "            project_dir=os.path.join(self.config.output_dir, \"logs\"),\n",
     "        )\n",
+    "        print(\"self.accelerator.is_main_process:\", self.accelerator.is_main_process)\n",
     "        if self.accelerator.is_main_process:\n",
     "            if self.config.output_dir is not None:\n",
     "                os.makedirs(self.config.output_dir, exist_ok=True)\n",
@@ -566,7 +568,7 @@
      "text": [
       "Number of parameters for nn_model: 160234497\n",
       "---------------- num_image = 100 -----------------\n",
-      "run_name = 0709-1331\n",
+      "run_name = 0709-1355\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",
@@ -587,14 +589,15 @@
      "output_type": "stream",
      "text": [
       "params loaded: (100, 2)\n",
-      "images rescaled to [-1.0, 1.1254141330718994]\n",
-      "params rescaled to [0.0022036265313531977, 0.9978807793709957]\n"
+      "images rescaled to [-1.0, 1.1527893543243408]\n",
+      "params rescaled to [0.01349925332487425, 0.9922290052754472]\n",
+      "self.accelerator.is_main_process: True\n"
      ]
     },
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "ae9f12def1154f6cb1eb0fc8d1e1871c",
+       "model_id": "00624ec4da1045659dcdab30773ac988",
        "version_major": 2,
        "version_minor": 0
       },
@@ -608,7 +611,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "cae19ac5ef7a4c34b6b57a6478dc159d",
+       "model_id": "817bb9f1a9534a6d9aeb353178cbada7",
        "version_major": 2,
        "version_minor": 0
       },
@@ -622,7 +625,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "b448b32948894b3c8e8780f1b6e6bf58",
+       "model_id": "076537a3f8ae45a888d6c57f71cad35c",
        "version_major": 2,
        "version_minor": 0
       },
@@ -636,7 +639,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "0271ab7c081a43ebb830dcfd3db145c1",
+       "model_id": "4fdf9b37d0f049538be616bea1c1ee15",
        "version_major": 2,
        "version_minor": 0
       },
@@ -650,7 +653,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "1bdf4c28272840f496288545bfdbdb96",
+       "model_id": "92922681843d42b2a0d56f67eaa32506",
        "version_major": 2,
        "version_minor": 0
       },
@@ -664,7 +667,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "01b50340758b4b05891e51a616660eb8",
+       "model_id": "8bf7c44087e244da85573e2146f18c74",
        "version_major": 2,
        "version_minor": 0
       },
@@ -678,7 +681,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "20ba64496e5e4467b97428cf6dcdbeb5",
+       "model_id": "bbeb4c2aa6f742e8bda3176f53f1b0d8",
        "version_major": 2,
        "version_minor": 0
       },
@@ -692,7 +695,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "8023e0bde0c3438fb218126c58fee954",
+       "model_id": "6d5b5b24015247c48544e995dfa917ec",
        "version_major": 2,
        "version_minor": 0
       },
@@ -706,7 +709,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "22881fbe8aff4ac1a3bde8735ab4fd24",
+       "model_id": "d0e3399ca4a54a9bae5e63c11bbd361f",
        "version_major": 2,
        "version_minor": 0
       },
@@ -720,7 +723,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "3bf15c9ba5144fa69fdddae271c7dbee",
+       "model_id": "a9939b9288c84fd99091702efe3a2e8a",
        "version_major": 2,
        "version_minor": 0
       },
@@ -849,6 +852,24 @@
     "#     # plt.imshow(images[0,0])\n",
     "#     # plt.show()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "1.12.0\n"
+     ]
+    }
+   ],
+   "source": [
+    "import torch\n",
+    "print(torch.__version__)"
+   ]
   }
  ],
  "metadata": {