随笔- 514 文章- 0 评论- 29 阅读- 139万

03 2023 档案

摘要：

t o r c h v i s i o n . t r a n s f o r m s . f u n c t i o n a l . p y

$torchvision.transforms.functional.py$ 中to_tensor()函数源码： def to_tensor(pic) -> Tensor: """Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor. This 阅读全文

posted @ 2023-03-20 10:59 做梦当财神阅读(156) 评论(0) 推荐(0) 编辑

with torch.no_grad()

摘要：讲述with torch.no_grad()前，先讲述requires_grad参数。 1. requires_grad 在

t o r c h

$torch$ 中，

t e n s o r

$tensor$ 有一个 requires_grad参数，如果设置为

T r u e

$True$ ，则反向传播时，该

t e n s o r

$tensor$ 就会自动求导。

t e n s o r

$tensor$ 阅读全文

posted @ 2023-03-06 23:46 做梦当财神阅读(222) 评论(0) 推荐(0) 编辑

torch.detach()、torch.detach_()

摘要：训练网络的时候希望保持一部分网络参数不变，只对其中一部分的参数进行调整；或训练部分分支网络，并不让其梯度对主网络的梯度造成影响，这时可以使用detach()切断一些分支的反向传播。 1. tensor.detach() 返回一个新的

t e n s o r

$tensor$ ，从当前计算图中分离下来，但仍指向原 $tenso 阅读全文

posted @ 2023-03-06 23:14 做梦当财神阅读(347) 评论(0) 推荐(0) 编辑

公告

昵称：做梦当财神
园龄： 7年5个月
粉丝： 115
关注： 28

+加关注

做梦当财神

03 2023 档案

公告

我的标签

积分与排名

随笔档案

阅读排行榜

推荐排行榜