|
from model.conv.DepthwiseSeparableConvolution import DepthwiseSeparableConvolution
|
|
import torch
|
|
from torch import nn
|
|
from torch.nn import functional as F
|
|
|
|
input=torch.randn(1,3,224,224)
|
|
dsconv=DepthwiseSeparableConvolution(3,64)
|
|
out=dsconv(input)
|
|
print(out.shape) |