Avoid dependency of imageio and Pytorch > 1.2 - Githubissues

amjltc295 / pytorch-golden-template

PyTorch Golden Template (under development)

13 stars 4 forks source link

Avoid dependency of imageio and Pytorch > 1.2 #67

Closed amjltc295 closed 4 years ago

amjltc295 commented 4 years ago

Why do we need this PR?

As a template, we should minimize the package dependency.
Also fix #37

How is it implemented?

Since imageio and inception is only needed in the FID calculation, it will now only be used when imported.
For Tensorboard summary writer, TensorboardX will be used when build-in Pytorch is not available.

PR readiness checklist

[x] Did it pass the Flake8 check?

[x] Did you test this PR?