/test-gpu-pytorch-old

Mock Weights & Biases training project that uses a NVIDIA GPU.

Primary LanguagePython

Test Repo

This repo runs a mock Weights & Biases training project that uses a NVIDIA GPU.

Dependency Version
PyTorch 1.4
CUDA 10.0

The entry point for this repo is train.py