Fixed the method `build_model` in the class `Solver`

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

2.66k stars 594 forks source link

Fixed the method `build_model` in the class `Solver` #52

Closed imadtoubal closed 1 year ago

imadtoubal commented 4 years ago

Fixed the issue where the function always created a model with 3 input channels and 1 output channel regardless of the config object values for img_ch and output_ch