Skip to content

Commit 63ff8c4

Browse files
hariharan-devarajanhariharan-devarajan
authored
Update megatron_deepspeed.yaml
1 parent baf023f commit 63ff8c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎dlio_benchmark/configs/workload/megatron_deepspeed.yaml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# 8 node run with 4 GPUs per node and TPSIZE=4 and PPSIZE=8
2-
model: unet3d
2+
model: megatron_deepspeed
33

44
framework: pytorch
55

0 commit comments

Comments
 (0)