senwu / emmental

A deep learning framework for building multimodal multi-task learning systems.
https://emmental.readthedocs.io
MIT License
108 stars 18 forks source link

Optimzer load cpu #95

Closed lorr1 closed 3 years ago

lorr1 commented 3 years ago

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 to torch.load

codecov[bot] commented 3 years ago

Codecov Report

Merging #95 (72931bd) into master (7ecaca2) will not change coverage. The diff coverage is n/a.

Impacted file tree graph

@@           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% <ø> (ø)