TORCHSCRIPT

이 튜토리얼은 C++와 같은 고성능 환경에서 실행될 수 있는 PyTorch 모델( nn.Module 의 하위클래스)의 중간 표현인 TorchScript에 대한 소개입니다.

Model Implementation

Sequential

~~~ x_train=tf.random.normal(shape=(1000,1),dtype=tf.float32) y_train=3x_train+1+0.2tf.random.normal(shape=(1000,1),dtype=tf.float32)

Pagination


© 2021.11. by zziny

Powered by zziny