wangkuiyi / gotorch

A Go idiomatic binding to the C++ core of PyTorch
MIT License
316 stars 35 forks source link

Reduce default number of threads to speed up training on GPUs #328

Closed shendiaomo closed 4 years ago

shendiaomo commented 4 years ago

Fix #321 , test on P100 before merge.

codecov[bot] commented 4 years ago

Codecov Report

Merging #328 into develop will increase coverage by 0.02%. The diff coverage is 100.00%.

Impacted file tree graph

@@             Coverage Diff             @@
##           develop     #328      +/-   ##
===========================================
+ Coverage    87.26%   87.29%   +0.02%     
===========================================
  Files           33       33              
  Lines         1445     1448       +3     
===========================================
+ Hits          1261     1264       +3     
  Misses         121      121              
  Partials        63       63              
Impacted Files Coverage Δ
tensor_gc.go 100.00% <ø> (ø)
device.go 100.00% <100.00%> (ø)