search

rlworkgroup / garage

A toolkit for reproducible reinforcement learning research.
MIT License
1.86k stars 309 forks source link

Move computation to gpu device #2205

Closed irisliucy closed 3 years ago

irisliucy commented 3 years ago

Addressing issue #2181 by moving computations to gpu device(s) set by global_device()

codecov[bot] commented 3 years ago

Codecov Report

Merging #2205 (cceb221) into master (43e7b78) will decrease coverage by 0.64%. The diff coverage is n/a.

Impacted file tree graph

@@            Coverage Diff             @@
##           master    #2205      +/-   ##
==========================================
- Coverage   91.84%   91.19%   -0.65%     
==========================================
  Files         242      201      -41     
  Lines       17456    10977    -6479     
  Branches     2537     1371    -1166     
==========================================
- Hits        16032    10011    -6021     
+ Misses       1061      704     -357     
+ Partials      363      262     -101
Impacted Files Coverage Δ
src/garage/sampler/utils.py 0.00% <0.00%> (-77.05%) :arrow_down:
src/garage/torch/algos/pearl.py 48.64% <0.00%> (-31.00%) :arrow_down:
src/garage/envs/multi_env_wrapper.py 79.51% <0.00%> (-14.60%) :arrow_down:
src/garage/tf/policies/policy.py 77.77% <0.00%> (-13.53%) :arrow_down:
src/garage/sampler/env_update.py 82.92% <0.00%> (-13.08%) :arrow_down:
src/garage/tf/algos/erwr.py 44.44% <0.00%> (-12.70%) :arrow_down:
src/garage/experiment/task_sampler.py 75.86% <0.00%> (-7.58%) :arrow_down:
...rage/tf/optimizers/conjugate_gradient_optimizer.py 81.90% <0.00%> (-7.21%) :arrow_down:
src/garage/np/_functions.py 77.39% <0.00%> (-6.30%) :arrow_down:
src/garage/tf/_functions.py 68.85% <0.00%> (-5.98%) :arrow_down:
... and 149 more

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 43e7b78...cceb221. Read the comment docs.