search

wolverinn / HEVC-CU-depths-prediction-CNN

Using convolutional neural networks to predict the Coding Units (CUs) depths in HEVC intra-prediction mode, in order to reduce the time of the encoding process in HEVC.
MIT License
78 stars 27 forks source link

Batch_size #1

Closed zhaorifa666 closed 4 years ago

zhaorifa666 commented 4 years ago

Hello,

Batch_size=1024 in your code, it is too large. Can you tell me the value of batch_size in your actual training model? Thank you very much!