Feat: Add AMP trainer and gradient accumulation for training optimization 3b8065d zhan1206 commited on Jun 5