6b430e8
1
2
import torch def f(*_):return(isinstance(_[0],torch.nn.Conv2d)and _[0].__dict__['kernel_size']==(1,1)and 128 in[_[0].__dict__['in_channels'],_[0].__dict__['out_channels']])