search

huawei-noah / vega

AutoML tools chain
http://www.noahlab.com.hk/opensource/vega/
Other
841 stars 177 forks source link

CARS for the Cutsom ClassificationDataset #283

Open XGBoost opened 1 year ago

XGBoost commented 1 year ago

The image size is 32. When the batch size is 1, the GPU memory is not overloaded, but when the batch size is 4, the GPU memory is overloaded. The GPU memory size is 16G. What could be the possible reasons for this issue?