随笔分类 - pytorch
深入理解pytorch
摘要:通常我们在训练模型时可以使用很多不同的框架,比如有的同学喜欢用 Pytorch,有的同学喜欢使用 TensorFLow,也有的喜欢 MXNet,以及深度学习最开始流行的 Caffe等等,这样不同的训练框架就导致了产生不同的模型结果包,在模型进行部署推理时就需要不同的依赖库,而且同一个框架比如tens
阅读全文
摘要:参考:https://pytorch.org/tutorials/beginner/examples_autograd/two_layer_net_custom_function.html torch.autograd.Function given a random x, y, W1, W2. y
阅读全文

浙公网安备 33010602011771号