The current example templates in the project folder are nice, but simpler than most engineering/research projects will be.
Would it be possible to have a template with all batteries included that shows how to properly use:
The DataModule introduced in 0.9.0, which you likely want to use if your dataset does not support something like: dataset = MNIST('', train=True, download=True, transform=transforms.ToTensor())
The new Metric API introduced in 1.0.0 with self.log()
The to_torchscript function. Needing to have access to the code to restore models, can be troublesome if a script change was not git pushed, but still used to train a model. Or library mismatches. TorchScript makes sure that the saved model can be used to reproduce the results, with the added benefit of working in C++.
The current example templates in the
project
folder are nice, but simpler than most engineering/research projects will be.Would it be possible to have a template with all batteries included that shows how to properly use:
dataset = MNIST('', train=True, download=True, transform=transforms.ToTensor())
self.log()
to_torchscript
function. Needing to have access to the code to restore models, can be troublesome if a script change was not git pushed, but still used to train a model. Or library mismatches. TorchScript makes sure that the saved model can be used to reproduce the results, with the added benefit of working in C++.