wangkuiyi / gotorch

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

Fix validation transforms in ResNet example #365

Closed Yancey1989 closed 3 years ago

Yancey1989 commented 3 years ago

The transforms on the validation dataset are different from training data. c.f. https://github.com/pytorch/examples/blob/master/imagenet/main.py#L225

codecov[bot] commented 3 years ago

Codecov Report

Merging #365 into develop will not change coverage. The diff coverage is n/a.

Impacted file tree graph

@@           Coverage Diff            @@
##           develop     #365   +/-   ##
========================================
  Coverage    87.62%   87.62%           
========================================
  Files           33       33           
  Lines         1487     1487           
========================================
  Hits          1303     1303           
  Misses         121      121           
  Partials        63       63           
shendiaomo commented 3 years ago

The transforms on the validation dataset are different from training data. c.f. https://github.com/pytorch/examples/blob/master/imagenet/main.py#L225

Great. This may lead to unstable accuracy on the validation dataset.