今天读voxelmorph代码时看到下面2句,想不到还可以有这种用法,通过变量控制神经网络的函数名称。
conv_fn = getattr(nn, "Conv{0}d".format(dim))
bn_fn = getattr(nn, "BatchNorm{0}d".format(dim)
今天读voxelmorph代码时看到下面2句,想不到还可以有这种用法,通过变量控制神经网络的函数名称。
conv_fn = getattr(nn, "Conv{0}d".format(dim))
bn_fn = getattr(nn, "BatchNorm{0}d".format(dim)