search

YanFangCS / AINET

Other
23 stars 1 forks source link

Get an error while training #4

Open Adnan1986 opened 1 year ago

Adnan1986 commented 1 year ago

Hi,

Thanks for sharing your code. To run your code I have tried to follow your instruction: After running Prerequisites and Data preparation part, when I was trying to run tain.sh script with following addresses python main.py --data='/home/adnan/superpixel/BSR/BSDS500/data/images/test/' --savepath='./ckpt/' --workers $worker --input_img_height 208 --input_img_width 208 --print_freq 20 --gpu $gpu --batch-size 16 --suffix '_myTrain'

I got a syntax error File "main.py", line 118 save_path = os.path.abspath(args.savepath) + '/' + f'{args.dataset}{args.suffix}' #os.path.join(args.dataset, save_path + '_' + timestamp )

And the reason is {args.dataset}

I can see you import dataset at the module, however couldn't find it in directories.

wangyxxjtu commented 1 year ago

Hello, sorry for late reply. We import the 'datasets' module at the begining of main.py, where implements some data reading functions. 'args.dataset' specifies which dataset you use