Fix ZeRO-3 UnboundLocalError (#968)

Commit

4 years ago

Fix ZeRO-3 UnboundLocalError (#968) * Fix UnboundLocalError * Get full partition size