Lachlan mnist loading fix - Githubissues

StrongResearch / isc-demos

Deep learning examples for the Instant Super Computer

11 stars 0 forks source link

Lachlan mnist loading fix #65

Closed StrongLachlanM closed 10 months ago

StrongLachlanM commented 10 months ago

Description

added map location for torch.load to load checkpoint

Things done

General:

[x] Linting (ruff, black, isort)