```python EfficientNet.from_pretrained("efficientnet-b{}".format(bN), num_classes=6, in_channels=channels) ``` ``` TypeError: from_pretrained() got an unexpected keyword argument 'in_channels' ``` might have to do with the fact that there are 2 from_pretrained() methods? not exactly sure Edit: I made some changes in my fork, will post a PR once I have time to test it