Gadersd/whisper-burn

Use Burn's PyTorchFileRecoder to Directly Load PyTorch Model

antimora opened this issue · 0 comments

Currently, the process of loading a PyTorch model into Burn requires an extra step of converting the model to Burn's format. An improvement that could streamline this process is to utilize Burn's PyTorchFileRecoder to load the PyTorch model directly.

The PyTorchFileRecoder is a convenient tool provided by Burn that allows direct loading of a PyTorch model file (.pt) and saves it into Burn's format. This eliminates the need for a separate conversion step and simplifies the overall workflow.

For more detailed information and examples, please refer to the Burn documentation on importing PyTorch models: https://burn.dev/book/import/pytorch-model.html