Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions deepspeed/ops/transformer/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,6 @@ class DeepSpeedTransformerConfig(TransformerConfig):
Arguments:
batch_size: The maximum batch size used for running the kernel on each GPU

max_seq_length: The sequence-length of the model being trained with DeepSpeed

hidden_size: The hidden size of the transformer layer

intermediate_size: The intermediate size of the feed-forward part of transformer layer
Expand Down