Resource
PyTorch codebase for training and using GPUNet model
Use the NGC CLI to download:
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
Dockerfile | 1.07 KB | November 4, 2022 UTC | |
Dockerfile-triton | 2.46 KB | November 4, 2022 UTC | |
eval.py | 2.16 KB | November 4, 2022 UTC | |
LICENSE | 11.14 KB | November 4, 2022 UTC | |
LICENSE AGREEMENT | 3.4 KB | November 4, 2022 UTC | |
NOTICE | 1.07 KB | November 4, 2022 UTC | |
README.md | 29.75 KB | November 4, 2022 UTC | |
requirements.txt | 681 B | November 4, 2022 UTC | |
train.py | 53.28 KB | November 4, 2022 UTC | |
train.sh | 109 B | November 4, 2022 UTC |