2020
10-08
10-08
pytorch中的weight-initilzation用法
pytorch中的权值初始化官方论坛对weight-initilzation的讨论torch.nn.Module.apply(fn)torch.nn.Module.apply(fn)#递归的调用weights_init函数,遍历nn.Module的submodule作为参数#常用来对模型的参数进行初始化#fn是对参数进行初始化的函数的句柄,fn以nn.Module或者自己定义的nn.Module的子类作为参数#fn(Module->None)?functiontobeappliedtoeachsubmodule#Returns:self#Returntype:Module例子:defweights_...
继续阅读 >