weberwcwei / AODnet-by-pytorch

Image Dehaze, Pytorch, An All-in-One Network for Dehazing, AOD-Net
138 stars 28 forks source link

An confusing error in your code #1

Closed Yukun-Huang closed 5 years ago

Yukun-Huang commented 6 years ago

In train.py, line 110 : varIn, varTar = varTar.float(), varIn.float()

I think the location of varIn and varTar should be exchanged.

weberwcwei commented 5 years ago

Fixed

lewisyangliu commented 5 years ago

The same error occurs in line 134 remained unchanged.