pycharm中'SiLU is already registered in activation layer'
没有找到相应答案,求大家帮忙解决
from mmgen.apis import init_model, sample_unconditional_model
config_file = 'configs/styleganv2/stylegan2_c2_lsun-church_256_b4x8_800k.py'
checkpoint_file = 'https://download.openmmlab.com/mmgen/stylegan2/official_weights/stylegan2-church-config-f-official_20210327_172657-1d42b7d1.pth'
device = 'cuda:0'
model = init_model(config_file, checkpoint_file, device=device)
fake_imgs = sample_unconditional_model(model, 4)
这是运行代码