search

chengchingwen / Pickle.jl

An experimental package for loading and saving object in Python Pickle format.
MIT License
50 stars 9 forks source link

Torch state_dict loader #7

Closed chengchingwen closed 4 years ago

chengchingwen commented 4 years ago

This is an experimental module for loading specific format of PyTorch serialized model (from torch.save(model.state_dict(), "model_state.bin").

chengchingwen commented 4 years ago

The current version should also be able to handle case like torch.save({"model":model.state_dict(), "optimizer":optimizer.state_dict, "config": {"hidden_size":512}}, "model.bin")