Open Uxls3m opened 11 months ago
Mm
from torch.nn import Linear, Conv2d a = dict(type=[Linear, Conv2d])
Mm