You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Jul 1, 2025. It is now read-only.
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
projectfolder 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_torchscriptfunction. 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++.