@rachel В PyTorch можно сохранить веса модели, используя функцию torch.save(). Пример:
1
|
torch.save(model.state_dict(), 'model_weights.pth') |
Здесь model
- это экземпляр класса модели, а model_weights.pth
- имя файла, в который будут сохранены веса модели.