debug-temp / temp.py
3v324v23's picture
add script
0ff5b7f
raw
history blame contribute delete
109 Bytes
import torch
conv = torch.load("depthwise_conv.pt")
h = torch.load("hidden_states.pt")
o = conv(h)
print(o)