hu-po/test-gpu-pytorch-new

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

Test Repo

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

Dependency	Version
PyTorch	1.10
CUDA	11.1

The entry point for this repo is train.py