From 356c02cdef1db092e8bbbbf73a9e2f43ee045802 Mon Sep 17 00:00:00 2001
From: Marco Kuhlmann <marco.kuhlmann@liu.se>
Date: Tue, 11 Feb 2025 16:16:38 +0100
Subject: [PATCH] Fix the name of the shard directory
---
labs/lab3/nlp-lab3.ipynb | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/labs/lab3/nlp-lab3.ipynb b/labs/lab3/nlp-lab3.ipynb
index adf0550..186db20 100644
--- a/labs/lab3/nlp-lab3.ipynb
+++ b/labs/lab3/nlp-lab3.ipynb
@@ -76,7 +76,7 @@
"@dataclass\n",
"class TrainingConfig:\n",
" device: torch.device = torch.device(\"cuda\")\n",
- " shard_dir: str = \"data\"\n",
+ " shard_dir: str = \"/courses/TDDE09/labs/lab3/data\"\n",
"\n",
" # Training steps and data processing\n",
" n_steps: int = 4768\n",
@@ -210,7 +210,7 @@
"metadata": {},
"outputs": [],
"source": [
- "sum(s.numel() for s in shards(\"/courses/TDDE09/labs/lab3/data\"))"
+ "sum(s.numel() for s in shards(TrainingConfig().shard_dir))"
]
},
{
--
GitLab