Closed leyuan-sun closed 2 years ago
What should I set in epochs and batch_size ?
batch-size depends on how much GPU-RAM you have ;), something between 8-32 might be enough. Epoch something between 100-200.
What should I set in epochs and batch_size ?