kevinMEH

ToRGB() Transform

3

### 🚀 The feature Some datasets (like ImageNet) contain both RGB images and grayscale images; to make the images batchable, they all need to have the same channel dimension. The...

Avoid branching in BasicBlock and Bottleneck

4

Avoid branching during forward pass in BasicBlock and Bottleneck block for better performance. Completely backwards compatible; downsampled = nn.Identity(x) is equivalent to downsampled = x Should be always more performant,...