Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit 63dbb76

Browse files
author
Jules Pondard
committed
Add a file to quickly compare different set of options
These options must be given in their vector format. See my_utils for more info
1 parent 9a17600 commit 63dbb76

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
import numpy as np
2+
import time
3+
import utils
4+
import torch
5+
import torch.optim as optim
6+
import torch.nn as nn
7+
import torch.utils.data
8+
import torch.nn.functional as F
9+
import tensor_comprehensions as tc
10+
11+
set_options = [
12+
[1, 1, 0, 1, 2, 1, 3, 6, 8, 0, 3, 0, 2, 11, 9, 8, 2, 0, 0, 3, 0, 0, 1, 0, 1, 2]
13+
]
14+
15+
16+
(tc_code, tc_name, inp, init_input_sz) = utils.get_convolution_example()
17+
18+
NB_HYPERPARAMS, INIT_INPUT_SZ = utils.NB_HYPERPARAMS, utils.INIT_INPUT_SZ
19+
20+
for i in range(len(set_options)):
21+
opts = np.array(set_options[i])
22+
time = utils.evalTime(opts)
23+
print(time)

0 commit comments

Comments
 (0)