import torch def main(): vocab = model = torch.load('model/transformer.vocab.pt') model = torch.load('model/model_step_15000.pt') print(vocab) if __name__ == "__main__": main()
