mirror of
https://github.com/AntoineHX/BU_Stoch_pool.git
synced 2025-05-04 01:30:48 +02:00
Setup env
This commit is contained in:
parent
e2db0e6057
commit
39711548ce
4 changed files with 45 additions and 5 deletions
2
main.py
2
main.py
|
@ -49,7 +49,7 @@ checkpoint=False
|
|||
|
||||
# Data
|
||||
print('==> Preparing data..')
|
||||
dataroot="./data"
|
||||
dataroot="~/scratch/data" #"./data"
|
||||
download_data=False
|
||||
transform_train = [
|
||||
# transforms.RandomCrop(32, padding=4),
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue