Closed lorr1 closed 3 years ago
Merging #95 (72931bd) into master (7ecaca2) will not change coverage. The diff coverage is
n/a
.
@@ Coverage Diff @@
## master #95 +/- ##
=======================================
Coverage 85.99% 85.99%
=======================================
Files 39 39
Lines 1892 1892
Branches 406 406
=======================================
Hits 1627 1627
Misses 153 153
Partials 112 112
Flag | Coverage Δ | |
---|---|---|
unittests | 85.99% <ø> (ø) |
Flags with carried forward coverage won't be shown. Click here to find out more.
Impacted Files | Coverage Δ | |
---|---|---|
src/emmental/learner.py | 70.27% <ø> (ø) |
Description of the problems or issues
When loading optimizer from pth file, it is loaded to the CPU and not freed, taking up more memory. Loading it to CPU prevents this from happening.
Description of the proposed changes
Added
map_location
totorch.load