search

VITA-Group / CADTransformer

[CVPR 2022]"CADTransformer: Panoptic Symbol Spotting Transformer for CAD Drawings", Zhiwen Fan, Tianlong Chen, Peihao Wang, Zhangyang Wang
MIT License
68 stars 20 forks source link

Running out of cuda memory #3

Closed fredlumm closed 1 year ago

fredlumm commented 2 years ago

Follow the given commands in REAMDE, the code would run successfully, but will soon running out of all cuda memory, it seems the gpu memory is increasing after every few batch.

olbod commented 2 years ago

@fredlumm how did you get the data for training?

A971RM commented 1 year ago

To get the data is direct. From https://github.com/VITA-Group/CADTransformer/blob/main/preprocess/download_data.py

https://drive.google.com/uc?id=16McNNY_-Y2uVnq42ntZTdYKPWgOZxwp3 https://drive.google.com/uc?id=1xgLqcj91i13_3vhfsUYcRYh3PhFYB9LJ https://drive.google.com/uc?id=1Hc4-ggsUMoB_5uqJdqYRn9K73QS8rOgG

OR follow README.md python preprocess/download_data.py --data_save_dir xx

olbod commented 1 year ago

@fredlumm It works for me. Thank you!

WHQCHINA commented 1 year ago

To get the data is direct. From https://github.com/VITA-Group/CADTransformer/blob/main/preprocess/download_data.py

https://drive.google.com/uc?id=16McNNY_-Y2uVnq42ntZTdYKPWgOZxwp3 https://drive.google.com/uc?id=1xgLqcj91i13_3vhfsUYcRYh3PhFYB9LJ https://drive.google.com/uc?id=1Hc4-ggsUMoB_5uqJdqYRn9K73QS8rOgG

OR follow README.md python preprocess/download_data.py --data_save_dir xx

downloading train... An error has occurred using proxy: None HTTPSConnectionPool(host='drive.google.com', port=443): Max retries exceeded with url: /uc?id=16McNNY_-Y2uVnq42ntZTdYKPWgOZxwp3 (Caused by ProxyError('Cannot connect to proxy.', OSError('Tunnel connection failed: 504 Gateway Time-out'))) downloading val... An error has occurred using proxy: None HTTPSConnectionPool(host='drive.google.com', port=443): Max retries exceeded with url: /uc?id=1xgLqcj91i13_3vhfsUYcRYh3PhFYB9LJ (Caused by ProxyError('Cannot connect to proxy.', OSError('Tunnel connection failed: 504 Gateway Time-out'))) downloading test... An error has occurred using proxy: None HTTPSConnectionPool(host='drive.google.com', port=443): Max retries exceeded with url: /uc?id=1Hc4-ggsUMoB_5uqJdqYRn9K73QS8rOgG (Caused by ProxyError('Cannot connect to proxy.', OSError('Tunnel connection failed: 503 Service Unavailable')))

WHQCHINA commented 1 year ago

@fredlumm It works for me. Thank you!

how did you get the data ?

zhiwenfan commented 1 year ago

use: python preprocess/download_data.py --data_save_dir /ssd1/zhiwen/datasets/svg_raw