Skip to content

Commit 8ca36a1

Browse files
committed
restoring # MAGIC comments in front of ! command cells
1 parent 9ca6687 commit 8ca36a1

File tree

1 file changed

+24
-24
lines changed

1 file changed

+24
-24
lines changed

train_dolly.py

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -34,14 +34,14 @@
3434

3535
# COMMAND ----------
3636

37-
!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb -O /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \
38-
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcublas-dev-11-3_11.5.1.109-1_amd64.deb -O /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \
39-
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb -O /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \
40-
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcurand-dev-11-3_10.2.4.109-1_amd64.deb -O /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb && \
41-
dpkg -i /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \
42-
dpkg -i /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \
43-
dpkg -i /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \
44-
dpkg -i /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb
37+
# MAGIC !wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb -O /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \
38+
# MAGIC wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcublas-dev-11-3_11.5.1.109-1_amd64.deb -O /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \
39+
# MAGIC wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb -O /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \
40+
# MAGIC wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcurand-dev-11-3_10.2.4.109-1_amd64.deb -O /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb && \
41+
# MAGIC dpkg -i /tmp/libcusparse-dev-11-3_11.5.0.58-1_amd64.deb && \
42+
# MAGIC dpkg -i /tmp/libcublas-dev-11-3_11.5.1.109-1_amd64.deb && \
43+
# MAGIC dpkg -i /tmp/libcusolver-dev-11-3_11.1.2.109-1_amd64.deb && \
44+
# MAGIC dpkg -i /tmp/libcurand-dev-11-3_10.2.4.109-1_amd64.deb
4545

4646
# COMMAND ----------
4747

@@ -167,22 +167,22 @@
167167

168168
# COMMAND ----------
169169

170-
!deepspeed {num_gpus_flag} \
171-
--module training.trainer \
172-
--input-model {input_model} \
173-
--deepspeed {deepspeed_config} \
174-
--epochs 2 \
175-
--local-output-dir {local_output_dir} \
176-
--dbfs-output-dir {dbfs_output_dir} \
177-
--per-device-train-batch-size {batch_size} \
178-
--per-device-eval-batch-size {batch_size} \
179-
--logging-steps 10 \
180-
--save-steps 200 \
181-
--save-total-limit 20 \
182-
--eval-steps 50 \
183-
--warmup-steps 50 \
184-
--test-size 200 \
185-
--lr 5e-6
170+
# MAGIC !deepspeed {num_gpus_flag} \
171+
# MAGIC --module training.trainer \
172+
# MAGIC --input-model {input_model} \
173+
# MAGIC --deepspeed {deepspeed_config} \
174+
# MAGIC --epochs 2 \
175+
# MAGIC --local-output-dir {local_output_dir} \
176+
# MAGIC --dbfs-output-dir {dbfs_output_dir} \
177+
# MAGIC --per-device-train-batch-size {batch_size} \
178+
# MAGIC --per-device-eval-batch-size {batch_size} \
179+
# MAGIC --logging-steps 10 \
180+
# MAGIC --save-steps 200 \
181+
# MAGIC --save-total-limit 20 \
182+
# MAGIC --eval-steps 50 \
183+
# MAGIC --warmup-steps 50 \
184+
# MAGIC --test-size 200 \
185+
# MAGIC --lr 5e-6
186186

187187
# COMMAND ----------
188188

0 commit comments

Comments
 (0)