search

xuebinqin / BASNet

Code for CVPR 2019 paper. BASNet: Boundary-Aware Salient Object Detection
MIT License
1.35k stars 249 forks source link

How much gpu memory does this code need? #3

Closed i4yyds closed 5 years ago

i4yyds commented 5 years ago

My machine is 1080ti, which is running short of gpu memory.

xuebinqin commented 5 years ago

Sorry. My fault. I changed the batch_size_train to 16 to run another new experiment later. It should be 8. Please change the line 78 of basnet_train.py: batch_size_train = 16 to batch_size_train = 8. Or try to re-download the code(I have already modified that line).

On Sun, Jun 2, 2019 at 1:55 AM SoulCoderCN notifications@github.com wrote:

My machine is 1080ti, which is running short of gpu memory.

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/NathanUA/BASNet/issues/3?email_source=notifications&email_token=ADSGOROLMCH3MH3QNH7PF7LPYN4GVA5CNFSM4HSCDLLKYY3PNVWWK3TUL52HS4DFUVEXG43VMWVGG33NNVSW45C7NFSM4GXEHKLQ, or mute the thread https://github.com/notifications/unsubscribe-auth/ADSGOROY3XXFBTWPBUQVLDDPYN4GVANCNFSM4HSCDLLA .

-- Xuebin Qin PhD Candidate Department of Computing Science University of Alberta, Edmonton, AB, Canada Homepage:https://webdocs.cs.ualberta.ca/~xuebin/