LazyLinear模块的作用就是可以帮助我们实现惰性初始化参数,不必在定义线性层的时候同时指定输入维度和输出维度,只需要指定输出维度即可,对于输入维度采用自动推断的方式。
官方的例子如下所示:
https://pytorch.org/docs/stable/generated/torch.nn.modules.lazy.LazyModuleMixin.html#torch.nn.modules.lazy.LazyModuleMixin
网站公告:飞燕AI训练营
AI公开课:YOLO相关技术公开课
提交前先勾选