update query

jswortz · Sep 25, 2023 · be04b21 · be04b21
1 parent ff60f36
commit be04b21
Show file tree

Hide file tree

Showing 3 changed files with 1,376 additions and 55 deletions.
diff --git a/01-bq-data-prep.ipynb b/01-bq-data-prep.ipynb
@@ -901,7 +901,7 @@
     "    IFNULL(audio.tempo, 0.0) AS track_tempo_can,\n",
     "    IFNULL(audio.time_signature, 0) AS track_time_signature_can,\n",
     "    ARRAY(\n",
-    "      SELECT CAST(pos_can - pos_pl \n",
+    "      SELECT CAST(pos_can - pos_pl) AS FLOAT64\n",
     "    FROM\n",
     "      UNNEST(seed_playlist_tracks) t\n",
     "    WHERE \n",

diff --git a/04-custom-train-retrieval.ipynb b/04-custom-train-retrieval.ipynb
@@ -491,7 +491,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 26,
    "id": "651cc211-4660-406d-a586-5e60ebfeb805",
    "metadata": {},
    "outputs": [],
@@ -548,7 +548,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 27,
    "id": "354feb4e-43c7-4238-9d4c-8a37b6278a62",
    "metadata": {},
    "outputs": [
@@ -565,19 +565,6 @@
       "                              '--candidate_file_dir=ndr-v1-hybrid-vertex-bucket',\n",
       "                              '--candidate_files_prefix=data/v1/candidates',\n",
       "                              '--experiment_name=scale-training-v1',\n",
-      "                              '--experiment_run=run-20230925-203116',\n",
-      "                              '--num_epochs=15',\n",
-      "                              '--batch_size=4096',\n",
-      "                              '--embedding_dim=128',\n",
-      "                              '--projection_dim=32',\n",
-      "                              '--layer_sizes=[512,256,128]',\n",
-      "                              '--learning_rate=0.01',\n",
-      "                              '--valid_frequency=7',\n",
-      "                              '--valid_steps=20',\n",
-      "                              '--epoch_steps=2003',\n",
-      "                              '--distribute=single',\n",
-      "                              '--model_version=v1',\n",
-      "                              '--pipeline_version=v1',\n",
       "                              '--seed=1234',\n",
       "                              '--max_tokens=20000',\n",
       "                              '--tb_resource_name=projects/934903580331/locations/us-central1/tensorboards/1356559854163984384',\n",
@@ -768,7 +755,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 28,
    "id": "33a58184-e0c4-4403-ad80-da5fdb7dbaed",
    "metadata": {},
    "outputs": [
@@ -792,33 +779,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": 32,
    "id": "d432d552-4416-4f4c-964f-bcca57b60a07",
    "metadata": {},
    "outputs": [],
    "source": [
     "# %load_ext tensorboard\n",
-    "%reload_ext tensorboard"
+    "# %reload_ext tensorboard"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 25,
+   "execution_count": 31,
    "id": "171227db-d5f9-47e4-8b67-a528950233ab",
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "ERROR: Timed out waiting for TensorBoard to start. It may still be running as pid 23612."
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "outputs": [],
    "source": [
-    "%tensorboard --logdir=$TB_LOGS_PATH"
+    "# %tensorboard --logdir=$TB_LOGS_PATH"
    ]
   },
   {