Update rcan.py

Alleviate gradient explosion, but convergence is still difficult
own
kongdebug 3 years ago committed by GitHub
parent b12bc7bb34
commit 393e570a68
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 14
      paddlers/custom_models/gan/generators/rcan.py

@ -7,12 +7,14 @@ from .builder import GENERATORS
def default_conv(in_channels, out_channels, kernel_size, bias=True):
return nn.Conv2D(
in_channels,
out_channels,
kernel_size,
padding=(kernel_size // 2),
bias_attr=bias)
weight_attr = paddle.ParamAttr(initializer=paddle.nn.initializer.XavierUniform(),
need_clip =True)
return nn.Conv2D(in_channels,
out_channels,
kernel_size,
padding=(kernel_size // 2),
weight_attr=weight_attr,
bias_attr=bias)
class MeanShift(nn.Conv2D):

Loading…
Cancel
Save