Fixing trainer with lightning

[zhiqwang]

Fixing #60

[codecov]

Codecov Report

Merging #128 (e95e76b) into master (a7e31a0) will increase coverage by 0.02%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master     #128      +/-   ##
==========================================
+ Coverage   83.54%   83.56%   +0.02%     
==========================================
  Files           8        8              
  Lines         784      779       -5     
==========================================
- Hits          655      651       -4     
+ Misses        129      128       -1     

Flag	Coverage Δ
unittests	83.56% <100.00%> (+0.02%)	⬆️

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

Impacted Files	Coverage Δ
test/common_utils.py	53.21% <ø> (-0.18%)	⬇️
test/test_onnx.py	87.95% <ø> (ø)
test/test_data_pipeline.py	100.00% <100.00%> (ø)

---

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 a7e31a0...e95e76b. Read the comment docs.