Cool, let’s see how much memory we can use starting with batch_size=4.                                                                                                                                                                      