Upload 3 files

ag4masses-public.ipynb

@@ -1,218 +1,5 @@
 {
  "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 10,
-   "metadata": {
-    "executionInfo": {
-     "elapsed": 611,
-     "status": "ok",
-     "timestamp": 1733595497864,
-     "user": {
-      "displayName": "Tong Peng",
-      "userId": "14680520704856526492"
-     },
-     "user_tz": 300
-    },
-    "id": "-IHoHd-t5sLP"
-   },
-   "outputs": [],
-   "source": [
-    "import sys, os\n",
-    "\n",
-    "AG4MDIR='/home/user/app/aglib/ag4masses'\n",
-    "AGLIB=f'/home/user/app/aglib/'\n",
-    "AGDIR=f\"{AG4MDIR}/alphageometry\"\n",
-    "MELIAD_PATH=f\"{AGLIB}/meliad\"\n",
-    "DATA=f\"{AGLIB}/ag_ckpt_vocab\"\n",
-    "TESTDIR=f\"/data/ag4mtest\""
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {
-    "id": "ASgGFu0NYHUH"
-   },
-   "source": [
-    "# Download Files"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Run this cell to refresh code and get the latest versions\n",
-    "# AG4MDIR and MELIAD_PATH are in /kaggle/working and will be saved as outputs of the Notebook\n",
-    "# !rm -fr {AG4MDIR} {MELIAD_PATH}"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "executionInfo": {
-     "elapsed": 5,
-     "status": "ok",
-     "timestamp": 1733594216384,
-     "user": {
-      "displayName": "Tong Peng",
-      "userId": "14680520704856526492"
-     },
-     "user_tz": 300
-    },
-    "id": "GgR_vO8XX9Vr"
-   },
-   "outputs": [],
-   "source": [
-    "if not os.path.exists(MELIAD_PATH):\n",
-    "    !git clone https://github.com/google-research/meliad.git {MELIAD_PATH}\n",
-    "\n",
-    "if not os.path.exists(AG4MDIR):\n",
-    "    !git clone https://github.com/tpgh24/ag4masses.git {AG4MDIR}\n",
-    "\n",
-    "# Temporarily modified files, upload into dataset tmpfiles\n",
-    "# !cp /kaggle/input/tmpfiles/numericals.py {AGDIR}\n",
-    "# !cp /kaggle/input/tmpfiles/alphageometry.py {AGDIR}\n",
-    "\n",
-    "if not os.path.exists(TESTDIR):\n",
-    "    !mkdir {TESTDIR}"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "executionInfo": {
-     "elapsed": 40247,
-     "status": "ok",
-     "timestamp": 1733594312243,
-     "user": {
-      "displayName": "Tong Peng",
-      "userId": "14680520704856526492"
-     },
-     "user_tz": 300
-    },
-    "id": "gP4zAZh2MHcv",
-    "outputId": "4796397b-8952-411e-bd33-8fd813865735"
-   },
-   "outputs": [],
-   "source": [
-    "if not os.path.exists(DATA):\n",
-    "  # download data: vocabulary, trained model\n",
-    "  !mkdir {DATA}\n",
-    "\n",
-    "  ### Convoluted process. This does no work, seems due to Google Drive restrictions\n",
-    "  # !gdown --folder https://bit.ly/alphageometry/\n",
-    "  #\n",
-    "  ### First got file links from Google Drive web UI, under Share menu. It will download HTML files with download button. Extract URL from the file.\n",
-    "  ### For checkpoint_10999999, because the file is big, there is an additional step asking for confirmation, got the final URL from the 2nd HTML,\n",
-    "  ### Link constructed using AI from HTML form.\n",
-    "  # !gdown https://drive.google.com/file/d/1mRd6J0UkeWoFUjeVB7BQi5lVNLvPBe31/view?usp=drive_link -O {AGLIB}/ag_ckpt_vocab/geometry.757.vocab\n",
-    "  # !gdown https://drive.google.com/file/d/1t-r3KfU8aDbS1UHpdyM3LH21rwSCIXTz/view?usp=drive_link -O {AGLIB}/ag_ckpt_vocab/geometry.757.model\n",
-    "  # !gdown https://drive.google.com/file/d/1qXkmmgoJ8oTYJdFV1xw0xGPpQj6SyOYA/view?usp=drive_link -O {AGLIB}/ag_ckpt_vocab/checkpoint_10999999\n",
-    "\n",
-    "  !wget -O {DATA}/geometry.757.vocab \"https://drive.usercontent.google.com/uc?id=1mRd6J0UkeWoFUjeVB7BQi5lVNLvPBe31&export=download\"\n",
-    "  !wget -O {DATA}/geometry.757.model \"https://drive.usercontent.google.com/uc?id=1t-r3KfU8aDbS1UHpdyM3LH21rwSCIXTz&export=download\"\n",
-    "  !wget -O {DATA}/checkpoint_10999999 \"https://drive.usercontent.google.com/download?id=1qXkmmgoJ8oTYJdFV1xw0xGPpQj6SyOYA&export=download&confirm=t&uuid=ae22f4de-cb77-4145-af5f-8cfbb59e867e\"\n",
-    "\n",
-    "!ls -l {DATA}"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "# Setup Env"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "executionInfo": {
-     "elapsed": 14190,
-     "status": "ok",
-     "timestamp": 1733594230570,
-     "user": {
-      "displayName": "Tong Peng",
-      "userId": "14680520704856526492"
-     },
-     "user_tz": 300
-    },
-    "id": "X8Aj3G0neT6K",
-    "outputId": "9538ceba-8065-44d6-a32f-35127e5f2575"
-   },
-   "outputs": [],
-   "source": [
-    "# Python packages for AlphaGeometry\n",
-    "!pip cache purge\n",
-    "!pip install --upgrade pip\n",
-    "!pip install --upgrade packaging setuptools setuptools_scm wheel \n",
-    "!pip install --require-hashes --no-deps -r {AG4MDIR}/alphageometry/requirements.txt"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "executionInfo": {
-     "elapsed": 15694,
-     "status": "ok",
-     "timestamp": 1733594246256,
-     "user": {
-      "displayName": "Tong Peng",
-      "userId": "14680520704856526492"
-     },
-     "user_tz": 300
-    },
-    "id": "u9fuBSr2qEwN",
-    "outputId": "97bbce78-8b49-4d3b-a831-d188a4a9e536"
-   },
-   "outputs": [],
-   "source": [
-    "# Python packages for Nvidia gpu.\n",
-    "# The versions of Python packages used by AlphaGeometry seem to only work with Cuda 11, not 12\n",
-    "!pip install -U \"jax==0.4.6\" \"jaxlib[cuda11_cudnn86]==0.4.6\" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\n",
-    "!pip install nvidia-cuda-runtime-cu11\n",
-    "!pip install nvidia-pyindex\n",
-    "# !pip list"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Linux packages for Nvidia gpu.\n",
-    "# The versions of Python packages used by AlphaGeometry seem to only work with Cuda 11, not 12\n",
-    "!apt-get update\n",
-    "!DEBIAN_FRONTEND=noninteractive apt install -y cuda-11-8\n",
-    "# !apt list|grep cuda"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "id": "fChy49CNhf01"
-   },
-   "outputs": [],
-   "source": [
-    "# Information about Nvidia drivers\n",
-    "!nvcc --version\n",
-    "!nvidia-smi"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -243,7 +30,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -251,6 +38,34 @@
     "PROB='imo-2024-q4'\n"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {
+    "executionInfo": {
+     "elapsed": 611,
+     "status": "ok",
+     "timestamp": 1733595497864,
+     "user": {
+      "displayName": "Tong Peng",
+      "userId": "14680520704856526492"
+     },
+     "user_tz": 300
+    },
+    "id": "-IHoHd-t5sLP"
+   },
+   "outputs": [],
+   "source": [
+    "import sys, os\n",
+    "\n",
+    "AG4MDIR='/home/user/app/aglib/ag4masses'\n",
+    "AGLIB=f'/home/user/app/aglib/'\n",
+    "AGDIR=f\"{AG4MDIR}/alphageometry\"\n",
+    "MELIAD_PATH=f\"{AGLIB}/meliad\"\n",
+    "DATA=f\"{AGLIB}/ag_ckpt_vocab\"\n",
+    "TESTDIR=f\"/data/ag4mtest\""
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {
@@ -262,7 +77,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -337,21 +152,25 @@
     "cd $AG4MDIR\n",
     "python -m alphageometry \\\n",
     "--alsologtostderr \\\n",
-    "--problems_file=$PROB_FILE \\\n",
+    "--problems_file=$PROBFILE \\\n",
     "--problem_name=$PROB \\\n",
     "--mode=$MODEL \\\n",
     "\"${DDAR_ARGS[@]}\" \\\n",
     "\"${SEARCH_ARGS[@]}\" \\\n",
     "\"${LM_ARGS[@]}\" \\\n",
     "--out_file=$OUTFILE \\\n",
-    "--n_workers=$NWORKERS # 2>&1\n",
+    "--n_workers=$NWORKERS 2>&1\n",
+    "\n",
+    "echo =======================================\n",
+    "echo Task Done.\n",
+    "echo See ag4mtest/$PROB.out and ag4mtest/$PROB.log for more information\n",
     "\n",
     "'''"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [
     {
@@ -362,6 +181,16 @@
       "+ ERRFILE=/data/ag4mtest/imo-2024-q4.log\n",
       "+ exec\n"
      ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "256"
+      ]
+     },
+     "execution_count": 7,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
@@ -405,61 +234,12 @@
     "\n"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 6,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "#!cat /kaggle/input/tmpfiles/test-problems.txt"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "# In an interactive Kaggle session, run this to see the log file. We can cancel this cell's execution\n",
-    "# to do other things in the Notebook\n",
-    "if os.environ[\"KAGGLE_KERNEL_RUN_TYPE\"] != \"Batch\":\n",
-    "    !tail -f {TESTDIR}/{PROB}.log"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Command to kill the background job in an interactive session\n",
-    "# !pkill -P `ps -ef|grep -- '-m alphageometry'|grep -v grep|awk '{printf \"%d,%d\", $2, $3}'`\n",
-    "\n",
-    "# Command to check processes\n",
-    "### |cat to show full commandline\n",
-    "# !ps -eo pid,ppid,pgid,%cpu,cmd |cat\n",
-    "# !ps -ef|cat"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Command to check progress of a running job in an interactive session\n",
-    "# !bash {AG4MDIR}/utils/checkprog.sh {TESTDIR}/{PROB}.log"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# In Batch run, after the job completes, list output files\n",
-    "!ls -ltr {TESTDIR}"
-   ]
+   "source": []
   }
  ],
  "metadata": {

download.sh

@@ -22,6 +22,8 @@ export DATA=ag_ckpt_vocab
 
 cd /home/user/app
 # some patch for cpu
-cp models.py /home/user/app/aglib/ag4masses/alphageometry/models.py
-cp lm_inference.py /home/user/app/aglib/ag4masses/alphageometry/lm_inference.py
-cp ag4masses-public.ipynb /data/ag4masses-public.ipynb
+# cp models.py /home/user/app/aglib/ag4masses/alphageometry/models.py
+# cp lm_inference.py /home/user/app/aglib/ag4masses/alphageometry/lm_inference.py
+cp ag4masses-public.ipynb /data/ag4masses-public.ipynb
+cp problems.txt /data/problems.txt
+mkdir /data/ag4mtest

problems.txt

@@ -0,0 +1,44 @@
+angbisec_bline_cyclic
+a b c = triangle; f = on_bline f b c, angle_bisector f b a c ? cyclic a b c f
+9point_part1
+a b c = triangle a b c; d = foot d a b c; e = midpoint e b a; f = midpoint f c b; g = midpoint g a c ? cyclic d g e f
+9point_part2
+a b c = triangle a b c; o = orthocenter o a b c; e = midpoint e b a; f = midpoint f c b; g = midpoint g a c; h = midpoint h a o ? cyclic e f g h
+castillon_4p
+a b c d = quadrangle; o z = segment; x = on_line a z, on_circle o z; u = on_circum z b x, on_line a b; y = on_line d z, on_circle o z; v = on_circum z c y, on_line c d; w = on_line a b, on_line c d; p = on_circum w u v, on_circle o z; q = on_line p a, on_circle o z; s = on_line p d, on_circle o z; r = on_line q b, on_line s c ? cong o z o r
+castillon_lemma_ddar_ok
+a c = segment; b = on_circle b a c; d = on_line a c, lc_tangent d b a; e = foot e b a c; f = on_circle f a c; g = on_tline g d a c, lc_tangent g f a; h = on_line d g, on_line e f; i = on_circle a c; j = on_circle a c, on_line i g; k = on_circle a c, on_line i h ? coll e j k
+napoleon_ddar_ok
+a1 a2 a3 = triangle; b3 = on_circle a1 a2, s_angle a1 a2 b3 60; b1 = on_circle a2 a3, s_angle a2 a3 b1 60; b2 = on_circle a3 a1, s_angle a3 a1 b2 60; c1 = circumcenter b1 a2 a3; c2 = circumcenter b2 a3 a1; c3 = circumcenter b3 a1 a2 ? cong c1 c2 c1 c3
+napoleon2
+a1 a2 a3 = triangle; c3 = s_angle a1 a2 c3 30, s_angle a2 a1 c3 150; c1 = s_angle a2 a3 c1 30, s_angle a3 a2 c1 150; c2 = s_angle a3 a1 c2 30, s_angle a1 a3 c2 150 ? cong c1 c2 c1 c3
+ceva
+a b c = triangle; d = free; e = on_line b c, on_line a d; f = on_line c a, on_line b d; g = on_line a b, on_line c d; h = on_line c a, on_pline h b e f ? eqratio f h f a g b g a
+ceva_hint_ddar_ok
+a b c = triangle; d = free; e = on_line b c, on_line a d; f = on_line c a, on_line b d; g = on_line a b, on_line c d; h = on_line c a, on_pline h b e f; i = on_line b d, on_pline i a b c; j = on_line a i, on_line c d ? eqratio f h f a g b g a
+ceva_menelaus_hint_ddar_ok
+a b c = triangle; d = free; e = on_line b c, on_line a d; f = on_line c a, on_line b d; g = on_line a b, on_line c d; h = on_line c a, on_pline h b e f; i = on_line c d, on_pline i e a b; j = on_line b f, on_pline j e a c ? eqratio f h f a g b g a
+square_angle15
+a b c d = isquare a b c d; e = s_angle c d e 15, s_angle d c e -15 ? cong e a a b
+5circles_ddar_ok
+p1 p2 p3 p4 p5 = pentagon; q1 = on_line p2 p5, on_line p1 p3; q2 = on_line p1 p3, on_line p2 p4; q3 = on_line p2 p4, on_line p3 p5; q4 = on_line p3 p5, on_line p4 p1; q5 = on_line p4 p1, on_line p2 p5; o1 = circumcenter p1 q1 q5; o2 = circumcenter p2 q1 q2; o3 = circumcenter p3 q2 q3; o4 = circumcenter p4 q3 q4; o5 = circumcenter p5 q4 q5; m1 = on_circle o1 q1, on_circle o2 q1; m2 = on_circle o2 q2, on_circle o3 q2; m3 = on_circle o3 q3, on_circle o4 q3; m4 = on_circle o4 q4, on_circle o5 q4; m5 = on_circle o5 q5, on_circle o1 q5 ? cyclic m1 m2 m3 m5
+butterfly
+a = free; b = free; c = on_circle a b; d = midpoint b c; e = on_circle a b; f = on_line d e, on_circle a b; g = on_circle a b; h = on_line d g, on_circle a b; i = on_line b c, on_line f g; j = on_line b c, on_line e h ? cong d i d j
+gaolian100-6-lemma_ddar_ok
+a = free; b = free; c = on_circle a b; d = on_circle a b; e = lc_tangent c a, lc_tangent d a; f = midpoint c d; g = on_line e b, on_circle a b ? eqangle f d f g f b f d
+gaolian100-6
+a = free; b = free; c = on_circle a b; d = on_circle a b; e = lc_tangent c a, lc_tangent d a; f = foot d a b; g = on_line d f, on_line e b; h = on_line d f, on_line c b ? cong d g g h
+gaolian100-12
+a = free; b = free; c = midpoint a b; d = on_circle c a; e = on_circle c a; f = on_tline f b b a, on_line d e; g = on_line a e, on_line c f; h = on_line a d, on_line c f ? cong c g c h
+gaolian100-26
+a b c = triangle a b c; d = angle_bisector d b a c, on_circum d a b c; e = circumcenter e a b c; f = on_line f a b, on_pline f e b d; g = on_line g a c, on_pline g e c d; h = orthocenter h a b c; i = on_line i b c, on_pline i h a d ? cong f b f i
+gaolian100-26-hint_ddar_ok
+a b c = triangle a b c; d = angle_bisector d b a c, on_circum d a b c; e = circumcenter e a b c; f = on_line f a b, on_pline f e b d; g = on_line g a c, on_pline g e c d; h = orthocenter h a b c; i = on_line i b c, on_pline i h a d; j = on_line e c, eqdistance j e e c ? cong f b f i
+gaolian100-98
+a b c d = quadrangle; e = on_line b c; f = on_line c d, on_aline f a d b a e; g = on_line b f, on_line d e ? eqangle a e a c a g a f
+pascal
+a = free; b = free; c = on_circle a b; d = on_circle a b; e = on_circle a b; f = on_circle a b; g = on_circle a b; h = intersection_ll b c e f; i = intersection_ll c d f g; j = intersection_ll d e g b ? coll h i j
+pascal_3p
+a = free; b = free; c = on_circle a b; d = on_circle a b; e = lc_tangent b a, on_line c d; f = lc_tangent c a, on_line d b; g = lc_tangent d a, on_line b c ? coll e f g
+imo-2024-q4
+a b c = triangle; i = incenter a b c; f = foot i a c; g = foot i a b; m = on_line i f, on_circle i f; n = on_line i g, on_circle i g; x = on_line b c, on_pline x m a c; y = on_line b c, on_pline y n a b; k = midpoint a c; l = midpoint a b; p = on_line a i, on_circum a b c ? eqangle p y p x i l i k