about batch_size - Githubissues

gcucurull / visual-compatibility

Context-Aware Visual Compatibility Prediction (https://arxiv.org/abs/1902.03646)

115 stars 37 forks source link

Open outfit-net opened 3 years ago

outfit-net commented 3 years ago

I run the program and found an error indicating that I was out of memory I would like to ask how to adjust the batch_size of this network