Efficient-PyTorch:我使用PyTorch训练大型数据集的最佳实践

时间:2024-06-02 08:37:42
【文件属性】:

文件名称:Efficient-PyTorch:我使用PyTorch训练大型数据集的最佳实践

文件大小:10KB

文件格式:ZIP

更新时间:2024-06-02 08:37:42

Python

高效火炬 我使用PyTorch训练大型数据集的最佳实践。 速度概述 通过遵循这些提示,当在ImageNet上训练ResNet-50时,使用PyTorch可以达到约730幅图像/秒。 根据和上的基准测试报告,该性能仍然具有竞争力。 Epoch: [0][430/5005] Time 0.409 (0.405) Data 626.6 (728.0) Loss 6.8381 (6.9754) Error@1 100.000 (99.850) Error@5 99.609 (99.259) Epoch: [0][440/5005] Time 0.364 (0.404) Data 704.2 (727.9) Loss 6.8506 (6.9725) Error@1 100.000 (99.851) Error@5 99.609 (99


【文件预览】:
Efficient-PyTorch-master
----.gitignore(1KB)
----requirements.txt(31B)
----tools()
--------README.md(1KB)
--------folder2lmdb.py(6KB)
----README.md(4KB)
----tfpack2lmdb.py(963B)
----main.py(12KB)

网友评论