随笔分类 -  pytorch

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