giant / src /deps.py
edgetensor's picture
Initial commit with folder contents
6b430e8 verified
raw
history blame contribute delete
172 Bytes
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']])