Support training with vanilla module

[zhiqwang]

Closes #86

[codecov]

Codecov Report

Merging #87 (5495d7f) into master (c4d3962) will increase coverage by 1.33%.
The diff coverage is 96.92%.

@@            Coverage Diff             @@
##           master      #87      +/-   ##
==========================================
+ Coverage   81.97%   83.31%   +1.33%     
==========================================
  Files           8        9       +1     
  Lines         738      797      +59     
==========================================
+ Hits          605      664      +59     
  Misses        133      133              

Flag	Coverage Δ
unittests	83.31% <96.92%> (+1.33%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
test/test_torchscript.py	100.00% <ø> (+2.00%)	⬆️
test/test_engine.py	99.23% <96.77%> (+0.19%)	⬆️
test/test_data_pipeline.py	97.05% <97.05%> (ø)

---

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 c4d3962...5495d7f. Read the comment docs.