3v324v23 commited on
Commit
0ff5b7f
1 Parent(s): 83202a6

add script

Browse files
Files changed (1) hide show
  1. temp.py +6 -0
temp.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ import torch
2
+
3
+ conv = torch.load("depthwise_conv.pt")
4
+ h = torch.load("hidden_states.pt")
5
+ o = conv(h)
6
+ print(o)