Fixed an issue where squeeze made the tensor dimension wrong

yihuacheng / IVGaze

What Do You See in Vehicle? Comprehensive Vision Solution for In-Vehicle Gaze Estimation

Apache License 2.0

26 stars 3 forks source link

Open CrossEntropy opened 2 months ago

CrossEntropy commented 2 months ago

Fixed dimension error during squeeze operation in 1x1 convolution #4