mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
35becd1879
Summary: Considered four different existing formats: 1) static graph, 2) torch script, 3) pickle files, 4) PyTorch C++ serialize APIs Pull Request resolved: https://github.com/pytorch/pytorch/pull/12149 Reviewed By: BIT-silence Differential Revision: D10098106 Pulled By: houseroad fbshipit-source-id: 94ec7fc57c842e50fae5286ddeda657a4967a07a
16 lines
421 B
Python
16 lines
421 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
from caffe2.python import convert, workspace
|
|
from caffe2.proto import caffe2_pb2, torch_pb2
|
|
import unittest
|
|
import numpy as np
|
|
|
|
class TestOperator(unittest.TestCase):
|
|
def setUp(self):
|
|
workspace.ResetWorkspace()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|