/AnimalVision-DirectML

An Image Classification Project model built using PyTorch and DirectML backend. Secondary purpose is to draw attention to AMD+DirectML and perform benchmarks on AMD GPUs.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers