Closed zorian15 closed 3 years ago
Instead of returning the entire callable object of the torch.nn classes when calling model.str_summary(), just return the __name__.
torch.nn
model.str_summary()
__name__
Closes #135
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.
make test
#278
Description
Instead of returning the entire callable object of the
torch.nn
classes when callingmodel.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
.Checklist:
#278
) has been searched for in the code to find relevant notes