... init.calculate_gain("relu") self.conv1.weight.data.mul_(relu_gain) ... Project: pggan-pytorch Author: nashory File: custom_layers.py License: MIT ...
確定! 回上一頁