matsengrp / torchdms

Analyze deep mutational scanning data with PyTorch
https://matsengrp.github.io/torchdms/
2 stars 0 forks source link

135-model-name-cleanup #136

Closed zorian15 closed 3 years ago

zorian15 commented 3 years ago

Description

Instead of returning the entire callable object of the torch.nn classes when calling model.str_summary(), just return the __name__.

Closes #135

Tests

Please describe the tests added to verify correct behavior. No unit tests here -- but scatterplots look like this again after a good ole' make test.

Screen Shot 2021-10-08 at 10 56 30 AM

Checklist: