File size: 172 Bytes
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']])