人工智能(Artificial Intelligence,简称AI)是指计算机系统在完成类似人类智力所需的任务时所表现出来的能力。它是一种复杂的技术,通过将大量的数据输入到算法中进行学习,不断调整和改进自己的算法,从而不断优化其性能。
人工智能可以被分为两类:弱人工智能和强人工智能。
弱人工智能(也称为狭义人工智能)是指只能在特定的任务领域中表现出类似人类的智能的人工智能系统。例如,语音识别系统、自动驾驶系统等。而强人工智能(也称为广义人工智能)则是指一种能够像人类一样在各种任务领域中表现出类似人类的智能的人工智能系统。
import torch
class Model(torch.nn.Module):
def __init__(self,n):
super().__init__()
self.n=n
self.conv=torch.nn.Conv2d(3,3,3)
def forward(self,x):
for i in range(self.n):
x=self.conv(x)
return x
models=[Model(2),Model(3)]
model_names=['model_2','model_3']
for model,model_name in zip(models,model_names):
dummy_input=torch.rand(1,3,10,10)
dummy_output=model(dummy_input)
model_trace=torch.jit.trace(model,dummy_input)
model_script=torch.jit.script(model)
#跟踪法与直接torch.onnx.export(model,...)等价
torch.onnx.export(model_trace,dummy_input,f'{model_name}_trace.onnx',example_outputs=dummy_output)
#记录法必须先调用torch.jit.sciprt
torch.onnx.export(model_script,dummy_input,f'{model_name}_script.onnx',example_outputs=dummy_output)