Breaking News: Grepper is joining You.com. Read the official announcement!

torch how to save model as json file

A M answered on December 4, 2022 Popularity 2/10 Helpfulness 3/10

answer torch how to save model as json file

torch how to save model as json file

Comment

import torch
from torch.utils.data import Dataset

from json import JSONEncoder
import json

class EncodeTensor(JSONEncoder,Dataset):
    def default(self, obj):
        if isinstance(obj, torch.Tensor):
            return obj.cpu().detach().numpy().tolist()
        return super(NpEncoder, self).default(obj)

with open('torch_weights.json', 'w') as json_file:
    json.dump(model.state_dict(), json_file,cls=EncodeTensor)

Popularity 2/10 Helpfulness 3/10 Language javascript

Source: stackoverflow.com

Tags: file javascript json model save torch

Link to this answer
Share Copy Link