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