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

fixed problem with inference with MyLeNetNoceil

Browse source
This commit is contained in:
Marco Pedersoli 2020-06-30 11:56:51 -04:00
parent 019310e60c
commit 5504ca67ab
4 changed files with 157 additions and 138 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -51,7 +51,7 @@ checkpoint=False
# Data
print('==> Preparing data..')
dataroot="~/scratch/data" #"./data"
dataroot="./data"#"~/scratch/data" #"./data"
download_data=False
transform_train = [
# transforms.RandomCrop(32, padding=4),