特斯拉前AI總監(jiān)Karpathy瘋狂點贊:單張A100,10秒訓練CIFAR-10,破世界紀錄!

新智元報道

新智元報道
【新智元導讀】這位小哥在單張A100上,10秒完成了CIFAR-10訓練,打破了世界紀錄,還引來了Karpathy嘖嘖稱贊。

項目介紹

git clone https://github.com/tysam-code/hlb-CIFAR10 && cd hlb-CIFAR10 && python -m pip install -r requirements.txt && python main.py
簡單 初學者友好 只需torch和python的常用代碼 可破解 幾乎不用外部依賴(目前只有torch和torchvision) 破世界紀錄的單GPU訓練時間(此前在A100上的世界紀錄是18.1秒,而Tysam打破了世界紀錄——小于10秒) 在2年內讓訓練時間小于2秒

壓縮和激勵層(squeeze and excite layers) 非常多的超參數(shù)調整 復雜架構修剪(miscellaneous architecture trimmings) 更改內存格式,更好地使用張量核心等 其他
Karpathy瘋狂點贊





著名的CIFAR-10



評論
圖片
表情
