Skip to content

Commit

Permalink
Auto-commit updated notebooks
Browse files Browse the repository at this point in the history
  • Loading branch information
ctr26 committed Sep 17, 2024
1 parent 547ebf9 commit 8b2db8f
Showing 1 changed file with 21 additions and 4 deletions.
25 changes: 21 additions & 4 deletions notebooks/idr.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -249,7 +249,9 @@
"cell_type": "code",
"execution_count": null,
"id": "75a403b4",
"metadata": {},
"metadata": {
"lines_to_next_cell": 0
},
"outputs": [],
"source": [
"# Main function to submit the training job using SLURM\n",
Expand Down Expand Up @@ -279,14 +281,29 @@
"metadata": {},
"outputs": [],
"source": [
"# Entry point for running the training script directly\n",
" # Entry point for running the training script directly\n",
"\n",
"\n",
"if __name__ == \"__main__\":\n",
" # Use the FTP spec to train the model\n",
" spec = fsspec_ftp\n",
" # Start local training\n",
" train(spec)\n",
" train(spec)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "3871a16d",
"metadata": {},
"outputs": [],
"source": [
"# Entry point for running the training script directly\n",
"if __name__ == \"__main__\":\n",
" # Use the local spec to train the model\n",
" spec = fsspec_local\n",
" # Alternatively, submit the job to SLURM\n",
" slurm(spec)"
" # slurm(spec)"
]
}
],
Expand Down

0 comments on commit 8b2db8f

Please sign in to comment.