AntoineH/BU_Stoch_pool
1
0
Fork 0
mirror of https://github.com/AntoineHX/BU_Stoch_pool.git synced 2025-05-04 01:30:48 +02:00
Code Issues Projects Releases Packages Wiki Activity

Benchmark NoCeil

Browse source
This commit is contained in:
Antoine Harlé 2020-06-24 01:57:32 -07:00
parent 115a8d80b5
commit 019310e60c
83 changed files with 15790 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

9
main.py
View file

@ -38,7 +38,9 @@ parser.add_argument('--res_folder', '-rf', dest='res_folder', default='res/',
parser.add_argument('--postfix', '-pf', dest='postfix', default='',
help='Results postfix')
parser.add_argument('--dataset', '-d', dest='dataset', default='CIFAR10',
help='Dataset')
help='Dataset')
parser.add_argument('-k', dest='k', default=3, type=int,
help='K for noCeil')
args = parser.parse_args()
print(args)
@ -115,7 +117,10 @@ print('==> Building model..')
#storchastic Bottom-UP like fig.3 paper
# net = MyLeNetMatStochBU() # 10.5s - 45.3% #1.3GB
net=globals()[args.net]()
try:
net=globals()[args.net](k=args.k) #NoCeil nets
except:
net=globals()[args.net]()
#print(net)
net = net.to(device)
if device == 'cuda':