Is it avaliable to load model in .pt file which is made in ‘pytorch’ ?
And I wonder how and what I should change.
you can load using the torch.load API.
Once you load the model you can use this inside a LightningModule
.
class MyModel(LightningModule):
def __init__(self, model_path):
self.model = torch.load(model_path)
...