accelerate
ed1a3365 - fix module and optimizer parameter mismatch before prepare_tp_ (#3845)

Commit
62 days ago
fix module and optimizer parameter mismatch before prepare_tp_ (#3845) * Add debug function to detect address mismatch in optimizer and module Add DTensor type class in get_named_parameters if DTensor exists before _prepare_tp * fix format * fix DTensor check condition * fix get_named_parameters logic in _prepare_tp
Author
Parents
Loading