|  | 
 
 发表于 2025-1-14 14:19:51
|
显示全部楼层 
| 跟着我折腾: with tf.device (models_opt_device):
 if 'df' in archi_type:
 self.encoder = model_archi.Encoder(in_ch=input_ch, e_ch=e_dims, name='encoder')
 encoder_out_ch = self.encoder.get_out_ch()*self.encoder.get_out_res(resolution)**2
 
 self.inter = model_archi.Inter (in_ch=encoder_out_ch, ae_ch=ae_dims, ae_out_ch=ae_dims, name='inter')
 
 上面的代码中,in_ch=encoder_out_ch, ae_ch=ae_dims,,这个e_ch=ae_dims,的值,调到32,你会发现象极了
 | 
 |