search

SyneRBI / SIRF-Exercises

SIRF Training and demonstration material
http://www.ccpsynerbi.ac.uk
Apache License 2.0
17 stars 21 forks source link

DL notebook fixes #222

Closed Imraj-Singh closed 2 months ago

Imraj-Singh commented 3 months ago

Dealing with a couple errors when running without GPU. One being print(torch.cuda.get_device_name(0)) is called even when installed without cuda, the other checkpoint = torch.load('trained.torch_model') mapped to GPU.

Also removing requiresgrad() as per Georg's comment.

KrisThielemans commented 3 months ago

Just ping me or @evgueni-ovtchinnikov when you want this to be merged.

KrisThielemans commented 3 months ago

best to add a line to CHANGES.md

Imraj-Singh commented 2 months ago

@KrisThielemans added a line to changes.md think it's fine to merge. They're very minor changes