Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

Published 20 hours ago • verified publisher icon aladdinpersson

Fixes to RenNet model

Open erfanbyt opened this issue 11 months ago • 0 comments

The PR contains the following fixes:

  1. the Block class wasn't following Python naming style
  2. Block class uses an extra attribute, stride, and it is never used. It can safely be deleted.
  3. _make_layer function and ResNet class are taking Block class as input while there is no need for it and it is accessible to them and no need to be given as input.

Thanks

erfanbyt avatar Dec 16 '24 06:12 erfanbyt