Make the replace module more configurable #1366
DeepSpeedInferenceConfig
f36d35b3
epsilon -> layer_norm_eps
b4a717e2
DeepSpeedTransformerConfig
41220bc5
configurabale stochastic_mode
b0b0a093
Updated replace_module.py
1e203ac8
Merge branch 'master' into replace_policy_config
10305168
kamalkraj
deleted the replace_policy_config branch 3 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub