search

langmanbusi / Semantic-Aware-Low-Light-Image-Enhancement

Semantic-Aware LLIE. CVPR 2023
MIT License
196 stars 7 forks source link

鉴别器 #32

Closed bigbiglower closed 4 months ago

bigbiglower commented 4 months ago

你好，我想问一下将trainer_seg_hist_fuse.py中from model.D_Net import Discriminator as D_Net中Discriminator改为contextDiscriminator后，model_D = D_Net().cuda()此行代码会报错。请问一下怎么解决？ 微信截图_20240603095518

langmanbusi commented 4 months ago

contextDiscriminator实例化的时候需要这两个参数，要自己设定一下