search

kiharalab / ACC-UNet

ACC-UNet is A Completely Convolutional UNet model inspired from transformer-based UNets
GNU General Public License v3.0
103 stars 14 forks source link

acc-unet Model parameter anomaly #8

Open mamengqi1113 opened 6 months ago

mamengqi1113 commented 6 months ago

Hi! I used torchsummary.summary(model, input_shape) to check the acc-unet model and found that the Forward/backward pass size exception occurred in all three versions, so I would like to consult you. Thank you.

Total params: 15,403,911 Trainable params: 15,403,911 Non-trainable params: 0

Input size (MB): 0.01 Forward/backward pass size (MB): 412316860328.29 Params size (MB): 58.76 Estimated Total Size (MB): 412316860387.07

mamengqi1113 commented 6 months ago

I tested it and it seems like there is something wrong with MLFC.