search

pystiche / papers

Reference implementation and replication of prominent NST papers
BSD 3-Clause "New" or "Revised" License
4 stars 1 forks source link

move augmentation device autocast from generation to usage #240

Closed pmeier closed 3 years ago

pmeier commented 3 years ago

This supersedes #235. Basically we move the autocast from the generation of the parameters (generate_parameters) to the usage (compute_transformation and apply_transform). With that we have direct access to the input and thus the device and don't need the comparably slow inspect module to fetch it from the caller workspace.

codecov[bot] commented 3 years ago

Codecov Report

Merging #240 (6e64140) into sanakoyeu-2018 (0897067) will increase coverage by 0.0%. The diff coverage is 100.0%.

Impacted file tree graph

@@              Coverage Diff               @@
##           sanakoyeu-2018    #240   +/-   ##
==============================================
  Coverage            94.8%   94.8%           
==============================================
  Files                  47      47           
  Lines                2359    2352    -7     
==============================================
- Hits                 2238    2232    -6     
+ Misses                121     120    -1
Impacted Files Coverage Δ
...tiche_papers/sanakoyeu_et_al_2018/_augmentation.py 86.9% <100.0%> (+<0.1%) :arrow_up:

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 0897067...6e64140. Read the comment docs.