2_Training.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Computer Vision Nanodegree\n",
    "\n",
    "## Project: Image Captioning\n",
    "\n",
    "---\n",
    "\n",
    "In this notebook, you will train your CNN-RNN model.  \n",
    "\n",
    "You are welcome and encouraged to try out many different architectures and hyperparameters when searching for a good model.\n",
    "\n",
    "This does have the potential to make the project quite messy!  Before submitting your project, make sure that you clean up:\n",
    "- the code you write in this notebook.  The notebook should describe how to train a single CNN-RNN architecture, corresponding to your final choice of hyperparameters.  You should structure the notebook so that the reviewer can replicate your results by running the code in this notebook.  \n",
    "- the output of the code cell in **Step 2**.  The output should show the output obtained when training the model from scratch.\n",
    "\n",
    "This notebook **will be graded**.  \n",
    "\n",
    "Feel free to use the links below to navigate the notebook:\n",
    "- [Step 1](#step1): Training Setup\n",
    "- [Step 2](#step2): Train your Model\n",
    "- [Step 3](#step3): (Optional) Validate your Model"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id='step1'></a>\n",
    "## Step 1: Training Setup\n",
    "\n",
    "In this step of the notebook, you will customize the training of your CNN-RNN model by specifying hyperparameters and setting other options that are important to the training procedure.  The values you set now will be used when training your model in **Step 2** below.\n",
    "\n",
    "You should only amend blocks of code that are preceded by a `TODO` statement.  **Any code blocks that are not preceded by a `TODO` statement should not be modified**.\n",
    "\n",
    "### Task #1\n",
    "\n",
    "Begin by setting the following variables:\n",
    "- `batch_size` - the batch size of each training batch.  It is the number of image-caption pairs used to amend the model weights in each training step. \n",
    "- `vocab_threshold` - the minimum word count threshold.  Note that a larger threshold will result in a smaller vocabulary, whereas a smaller threshold will include rarer words and result in a larger vocabulary.  \n",
    "- `vocab_from_file` - a Boolean that decides whether to load the vocabulary from file. \n",
    "- `embed_size` - the dimensionality of the image and word embeddings.  \n",
    "- `hidden_size` - the number of features in the hidden state of the RNN decoder.  \n",
    "- `num_epochs` - the number of epochs to train the model.  We recommend that you set `num_epochs=3`, but feel free to increase or decrease this number as you wish.  [This paper](https://arxiv.org/pdf/1502.03044.pdf) trained a captioning model on a single state-of-the-art GPU for 3 days, but you'll soon see that you can get reasonable results in a matter of a few hours!  (_But of course, if you want your model to compete with current research, you will have to train for much longer._)\n",
    "- `save_every` - determines how often to save the model weights.  We recommend that you set `save_every=1`, to save the model weights after each epoch.  This way, after the `i`th epoch, the encoder and decoder weights will be saved in the `models/` folder as `encoder-i.pkl` and `decoder-i.pkl`, respectively.\n",
    "- `print_every` - determines how often to print the batch loss to the Jupyter notebook while training.  Note that you **will not** observe a monotonic decrease in the loss function while training - this is perfectly fine and completely expected!  You are encouraged to keep this at its default value of `100` to avoid clogging the notebook, but feel free to change it.\n",
    "- `log_file` - the name of the text file containing - for every step - how the loss and perplexity evolved during training.\n",
    "\n",
    "If you're not sure where to begin to set some of the values above, you can peruse [this paper](https://arxiv.org/pdf/1502.03044.pdf) and [this paper](https://arxiv.org/pdf/1411.4555.pdf) for useful guidance!  **To avoid spending too long on this notebook**, you are encouraged to consult these suggested research papers to obtain a strong initial guess for which hyperparameters are likely to work best.  Then, train a single model, and proceed to the next notebook (**3_Inference.ipynb**).  If you are unhappy with your performance, you can return to this notebook to tweak the hyperparameters (and/or the architecture in **model.py**) and re-train your model.\n",
    "\n",
    "### Question 1\n",
    "\n",
    "**Question:** Describe your CNN-RNN architecture in detail.  With this architecture in mind, how did you select the values of the variables in Task 1?  If you consulted a research paper detailing a successful implementation of an image captioning model, please provide the reference.\n",
    "\n",
    "**Answer:** \n",
    "\n",
    "The CNN-RNN architecture implemented for this project is similar to those presented in the literature  and recommended in this notebook \\[1-3\\].\n",
    "\n",
    "**Encoder model (CNN)**: I used the provided pretrained ResNet-50 network as the CNN encoder. ResNet-50 (introduced in [this paper](https://arxiv.org/pdf/1512.03385.pdf)) is a residual deep neural network with 50 layers, including convoltional layers and residual pass through layers. Pretained ResNet models are available on the [PyTorch website](https://pytorch.org/vision/stable/models.html). I added a fully connected layer after the pretrained network that outputs `embed_size` for our image embeddings. Additionally, I added a one-dimensional [batch normalization](https://pytorch.org/docs/stable/generated/torch.nn.BatchNorm1d.html) layer after the fully connected layer to make the network more robust during training.\n",
    "\n",
    "**Decoder model (RNN)**: I used a two-layer LSTM network as the RNN encoder network. The function of this network is to input image embeddings and output descriptive text captions. The first part of the network is an [embedding layer](https://pytorch.org/docs/stable/generated/torch.nn.Embedding.html), which is a lookup table that stores embeddings of a fixed dictionary and size. The image feature embeddings are concatinated into tensors and fed through the [LSTM](https://pytorch.org/docs/stable/generated/torch.nn.LSTM.html) network. I used two-layer stacked LSTM to improve the accuracy of the LSTM network. The LSTM also includes dropout followed by a seperate dropout layer, both with a dropout probability of 0.3. Lastly, the output of the LSTM is fed through a fully connected layer that inputs the `hidden_size` and outputs to `vocab_size`.\n",
    "\n",
    "**Hyperparameter Values**: I selected a `batch_size` of 64 because it was recommended in \\[2\\] and helps improve training performance. I chose a minimum word count threshold `vocab_threshold` of 5 because it empirically provides good results and is recommended in \\[3\\]. The `embed_size` and `hidden_size` values are 512 as recommended in \\[1\\]. This value provides a good trade-off that allows the network to accurately learn information from the dataset while limiting the number of unknowns in the network. The model is trained for 3 epochs (`num_epochs`) as recommended in this notebook. The training took approximately ~9 hours using the GPU provided in this workspace. The training results in this notebook are cut off in the middle of the second epoch. However, the *training_log.txt* provides a training log for all three epochs and demonstrates a reduction of the loss function and good training results.\n",
    "\n",
    "References:\n",
    "\n",
    "\\[1\\] [Vinyals, Oriol, et al. \"Show and Tell: A neural image caption generator.\" Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. 2015.](https://arxiv.org/pdf/1411.4555.pdf)\n",
    "\n",
    "\\[2\\] [Xu, Kelvin, et al. \"Show, Attend and Tell: Neural image caption generation with visual attention.\" International Conference on Machine Learning. PMLR, 2015.](https://arxiv.org/pdf/1502.03044.pdf)\n",
    "\n",
    "\\[3\\] [Vinyals, Oriol, et al. \"Show and Tell: Lessons learned from the 2015 mscoco image captioning challenge.\" IEEE Transactions on Pattern Analysis and Machine Intelligence 39.4 (2016): 652-663.](https://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=7505636)\n",
    "\n",
    "\n",
    "### (Optional) Task #2\n",
    "\n",
    "Note that we have provided a recommended image transform `transform_train` for pre-processing the training images, but you are welcome (and encouraged!) to modify it as you wish.  When modifying this transform, keep in mind that:\n",
    "- the images in the dataset have varying heights and widths, and \n",
    "- if using a pre-trained model, you must perform the corresponding appropriate normalization.\n",
    "\n",
    "### Question 2\n",
    "\n",
    "**Question:** How did you select the transform in `transform_train`?  If you left the transform at its provided value, why do you think that it is a good choice for your CNN architecture?\n",
    "\n",
    "**Answer:** I used the default transform provided in this notebook because it provides image scaling, random cropping, random vertical and horizontal flips, and color normalization. This transformation provides good performance and does not require improvement.\n",
    "\n",
    "### Task #3\n",
    "\n",
    "Next, you will specify a Python list containing the learnable parameters of the model.  For instance, if you decide to make all weights in the decoder trainable, but only want to train the weights in the embedding layer of the encoder, then you should set `params` to something like:\n",
    "```\n",
    "params = list(decoder.parameters()) + list(encoder.embed.parameters()) \n",
    "```\n",
    "\n",
    "### Question 3\n",
    "\n",
    "**Question:** How did you select the trainable parameters of your architecture?  Why do you think this is a good choice?\n",
    "\n",
    "**Answer:** Since the CNN is a pretrained ResNet-50 network, the parameters in this model do not need to be trained. Therefor, I selected the trainable parameters in the encoder network to be the linear layer (*encoder.embed*) and batch normalization (*encoder.bn*) layer parameters. Additionally, all of the decoder parameters are trainable parameters.\n",
    "\n",
    "\n",
    "### Task #4\n",
    "\n",
    "Finally, you will select an [optimizer](http://pytorch.org/docs/master/optim.html#torch.optim.Optimizer).\n",
    "\n",
    "### Question 4\n",
    "\n",
    "**Question:** How did you select the optimizer used to train your model?\n",
    "\n",
    "**Answer:** I used an [Adam optimizer](https://pytorch.org/docs/stable/generated/torch.optim.Adam.html#torch.optim.Adam) because it has given me good results training other CNN and RNN networks and it converged faster than [stochastic gradient descent (SGD)](https://pytorch.org/docs/stable/generated/torch.optim.SGD.html#torch.optim.SGD) with momentum in my experiments training the network. I used the default parameters including a learning rate (*lr*) of 0.001. Additionally, the Adam optimizer is used in \\[2\\]."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[nltk_data] Downloading package punkt to /root/nltk_data...\n",
      "[nltk_data]   Unzipping tokenizers/punkt.zip.\n",
      "Vocabulary successfully loaded from vocab.pkl file!\n",
      "loading annotations into memory...\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\n",
      "\n",
      "  0%|          | 0/414113 [00:00<?, ?it/s]\u001b[A\u001b[A\n",
      "\n",
      "  0%|          | 411/414113 [00:00<01:40, 4104.46it/s]\u001b[A\u001b[A\n",
      "\n",
      "  0%|          | 897/414113 [00:00<01:35, 4304.48it/s]\u001b[A\u001b[A\n",
      "\n",
      "  0%|          | 1391/414113 [00:00<01:32, 4475.40it/s]\u001b[A\u001b[A\n",
      "\n",
      "  0%|          | 1886/414113 [00:00<01:29, 4607.19it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|          | 2372/414113 [00:00<01:28, 4678.23it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|          | 2858/414113 [00:00<01:26, 4729.98it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|          | 3348/414113 [00:00<01:25, 4777.54it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|          | 3874/414113 [00:00<01:23, 4910.58it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|          | 4370/414113 [00:00<01:23, 4924.58it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|          | 4876/414113 [00:01<01:22, 4963.65it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|▏         | 5384/414113 [00:01<01:21, 4995.79it/s]\u001b[A\u001b[A\n",
      "\n",
      "  1%|▏         | 5877/414113 [00:01<01:24, 4806.61it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 6368/414113 [00:01<01:24, 4836.52it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 6869/414113 [00:01<01:23, 4885.30it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 7356/414113 [00:01<01:23, 4854.03it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 7865/414113 [00:01<01:22, 4920.20it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 8381/414113 [00:01<01:21, 4986.43it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 8883/414113 [00:01<01:21, 4995.32it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 9383/414113 [00:01<01:21, 4961.59it/s]\u001b[A\u001b[A\n",
      "\n",
      "  2%|▏         | 9886/414113 [00:02<01:21, 4980.69it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 10391/414113 [00:02<01:20, 4998.98it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 10891/414113 [00:02<01:20, 4981.02it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 11390/414113 [00:02<01:21, 4932.79it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 11884/414113 [00:02<01:21, 4910.70it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 12376/414113 [00:02<01:23, 4838.88it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 12863/414113 [00:02<01:22, 4846.67it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 13378/414113 [00:02<01:21, 4933.01it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 13872/414113 [00:02<01:21, 4932.35it/s]\u001b[A\u001b[A\n",
      "\n",
      "  3%|▎         | 14366/414113 [00:02<01:21, 4894.91it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▎         | 14857/414113 [00:03<01:21, 4897.39it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▎         | 15361/414113 [00:03<01:20, 4938.00it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 15856/414113 [00:03<01:20, 4936.09it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 16350/414113 [00:03<01:20, 4928.14it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 16843/414113 [00:03<01:20, 4923.49it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 17336/414113 [00:03<01:20, 4923.82it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 17835/414113 [00:03<01:20, 4939.90it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 18330/414113 [00:03<01:20, 4933.53it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▍         | 18837/414113 [00:03<01:19, 4971.48it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▍         | 19337/414113 [00:03<01:19, 4978.16it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▍         | 19852/414113 [00:04<01:18, 5028.07it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▍         | 20362/414113 [00:04<01:17, 5049.13it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▌         | 20868/414113 [00:04<01:17, 5050.84it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▌         | 21374/414113 [00:04<01:17, 5050.37it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▌         | 21880/414113 [00:04<01:17, 5045.63it/s]\u001b[A\u001b[A\n",
      "\n",
      "  5%|▌         | 22385/414113 [00:04<01:18, 4976.10it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▌         | 22899/414113 [00:04<01:17, 5023.97it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▌         | 23402/414113 [00:04<01:18, 4985.21it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▌         | 23916/414113 [00:04<01:17, 5028.75it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▌         | 24420/414113 [00:04<01:17, 5031.31it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▌         | 24924/414113 [00:05<01:17, 5031.20it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▌         | 25428/414113 [00:05<01:17, 5022.03it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▋         | 25934/414113 [00:05<01:17, 5032.74it/s]\u001b[A\u001b[A\n",
      "\n",
      "  6%|▋         | 26438/414113 [00:05<01:17, 5008.63it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 26939/414113 [00:05<01:17, 4972.65it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 27437/414113 [00:05<01:18, 4916.91it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 27940/414113 [00:05<01:18, 4947.54it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 28444/414113 [00:05<01:17, 4974.86it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 28942/414113 [00:05<01:17, 4968.47it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 29439/414113 [00:05<01:18, 4894.43it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 29929/414113 [00:06<01:20, 4760.97it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 30424/414113 [00:06<01:19, 4816.02it/s]\u001b[A\u001b[A\n",
      "\n",
      "  7%|▋         | 30929/414113 [00:06<01:18, 4881.40it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 31422/414113 [00:06<01:18, 4894.12it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 31928/414113 [00:06<01:17, 4939.92it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 32423/414113 [00:06<01:17, 4894.95it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 32913/414113 [00:06<01:17, 4887.55it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 33403/414113 [00:06<01:17, 4883.93it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 33899/414113 [00:06<01:17, 4906.40it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 34399/414113 [00:06<01:16, 4933.09it/s]\u001b[A\u001b[A\n",
      "\n",
      "  8%|▊         | 34893/414113 [00:07<01:17, 4916.93it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▊         | 35385/414113 [00:07<01:17, 4904.09it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▊         | 35877/414113 [00:07<01:17, 4908.54it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▉         | 36369/414113 [00:07<01:16, 4909.83it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▉         | 36880/414113 [00:07<01:15, 4966.69it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▉         | 37377/414113 [00:07<01:16, 4949.54it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▉         | 37873/414113 [00:07<01:16, 4912.31it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▉         | 38366/414113 [00:07<01:16, 4916.66it/s]\u001b[A\u001b[A\n",
      "\n",
      "  9%|▉         | 38878/414113 [00:07<01:15, 4974.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|▉         | 39378/414113 [00:07<01:15, 4980.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|▉         | 39888/414113 [00:08<01:14, 5015.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|▉         | 40390/414113 [00:08<01:14, 4989.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|▉         | 40899/414113 [00:08<01:14, 5016.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|▉         | 41408/414113 [00:08<01:14, 5036.51it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|█         | 41912/414113 [00:08<01:13, 5032.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|█         | 42416/414113 [00:08<01:14, 4999.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|█         | 42917/414113 [00:08<01:14, 4970.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 10%|█         | 43426/414113 [00:08<01:14, 5004.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 43932/414113 [00:08<01:13, 5018.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 44434/414113 [00:08<01:14, 4981.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 44946/414113 [00:09<01:13, 5019.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 45449/414113 [00:09<01:13, 5015.67it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 45951/414113 [00:09<01:13, 5011.16it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 46459/414113 [00:09<01:13, 5030.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█▏        | 46963/414113 [00:09<01:13, 5012.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█▏        | 47465/414113 [00:09<01:13, 4965.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 47990/414113 [00:09<01:12, 5045.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 48495/414113 [00:09<01:12, 5045.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 49000/414113 [00:09<01:12, 5028.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 49504/414113 [00:09<01:12, 5020.22it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 50028/414113 [00:10<01:11, 5081.57it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 50537/414113 [00:10<01:11, 5077.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 51047/414113 [00:10<01:11, 5082.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 12%|█▏        | 51556/414113 [00:10<01:11, 5069.78it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 52064/414113 [00:10<01:11, 5034.92it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 52569/414113 [00:10<01:11, 5039.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 53074/414113 [00:10<01:12, 4988.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 53579/414113 [00:10<01:12, 5005.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 54084/414113 [00:10<01:11, 5016.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 54586/414113 [00:11<01:11, 5017.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 55088/414113 [00:11<01:11, 4990.61it/s]\u001b[A\u001b[A\n",
      "\n",
      " 13%|█▎        | 55599/414113 [00:11<01:11, 5024.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▎        | 56102/414113 [00:11<01:11, 5006.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▎        | 56603/414113 [00:11<02:07, 2794.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▍        | 57087/414113 [00:11<01:51, 3199.16it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▍        | 57593/414113 [00:11<01:39, 3594.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▍        | 58095/414113 [00:11<01:30, 3929.31it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▍        | 58616/414113 [00:12<01:23, 4241.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▍        | 59121/414113 [00:12<01:19, 4454.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 14%|█▍        | 59617/414113 [00:12<01:17, 4594.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▍        | 60130/414113 [00:12<01:14, 4742.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▍        | 60648/414113 [00:12<01:12, 4865.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▍        | 61154/414113 [00:12<01:11, 4920.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▍        | 61658/414113 [00:12<01:11, 4902.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▌        | 62165/414113 [00:12<01:11, 4949.52it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▌        | 62666/414113 [00:12<01:10, 4955.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▌        | 63166/414113 [00:12<01:10, 4951.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▌        | 63669/414113 [00:13<01:10, 4974.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▌        | 64169/414113 [00:13<01:10, 4978.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▌        | 64687/414113 [00:13<01:09, 5035.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▌        | 65197/414113 [00:13<01:09, 5053.58it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▌        | 65704/414113 [00:13<01:09, 5007.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▌        | 66206/414113 [00:13<01:09, 5000.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▌        | 66707/414113 [00:13<01:09, 4999.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▌        | 67214/414113 [00:13<01:09, 5017.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▋        | 67730/414113 [00:13<01:08, 5057.02it/s]\u001b[A\u001b[A\n",
      "\n",
      " 16%|█▋        | 68236/414113 [00:13<01:08, 5042.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 68741/414113 [00:14<01:08, 5038.99it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 69252/414113 [00:14<01:08, 5058.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 69758/414113 [00:14<01:08, 5019.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 70270/414113 [00:14<01:08, 5048.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 70776/414113 [00:14<01:08, 5009.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 71279/414113 [00:14<01:08, 5015.58it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 71781/414113 [00:14<01:08, 4974.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 17%|█▋        | 72288/414113 [00:14<01:08, 5000.13it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 72795/414113 [00:14<01:08, 5019.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 73298/414113 [00:14<01:07, 5012.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 73801/414113 [00:15<01:07, 5017.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 74308/414113 [00:15<01:07, 5030.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 74828/414113 [00:15<01:06, 5079.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 75337/414113 [00:15<01:07, 5055.31it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 75843/414113 [00:15<01:07, 4990.44it/s]\u001b[A\u001b[A\n",
      "\n",
      " 18%|█▊        | 76343/414113 [00:15<01:07, 4973.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▊        | 76852/414113 [00:15<01:07, 5004.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▊        | 77353/414113 [00:15<01:07, 4993.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 77861/414113 [00:15<01:07, 5016.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 78378/414113 [00:16<01:06, 5059.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 78885/414113 [00:16<01:06, 5013.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 79387/414113 [00:16<01:07, 4964.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 79899/414113 [00:16<01:06, 5008.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 80401/414113 [00:16<01:06, 4988.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|█▉        | 80902/414113 [00:16<01:06, 4992.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|█▉        | 81402/414113 [00:16<01:07, 4960.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|█▉        | 81905/414113 [00:16<01:06, 4978.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|█▉        | 82413/414113 [00:16<01:06, 5007.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|██        | 82919/414113 [00:16<01:05, 5021.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|██        | 83422/414113 [00:17<01:06, 5009.24it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|██        | 83927/414113 [00:17<01:05, 5021.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 20%|██        | 84430/414113 [00:17<01:05, 5001.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 84943/414113 [00:17<01:05, 5038.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 85447/414113 [00:17<01:05, 4992.29it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 85947/414113 [00:17<01:06, 4967.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 86453/414113 [00:17<01:05, 4993.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 86975/414113 [00:17<01:04, 5059.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 87482/414113 [00:17<01:05, 5024.57it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██        | 87985/414113 [00:17<01:05, 5004.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██▏       | 88491/414113 [00:18<01:04, 5019.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 21%|██▏       | 89009/414113 [00:18<01:04, 5063.00it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 89529/414113 [00:18<01:03, 5102.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 90040/414113 [00:18<01:04, 5032.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 90544/414113 [00:18<01:04, 4978.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 91043/414113 [00:18<01:04, 4977.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 91555/414113 [00:18<01:04, 5017.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 92062/414113 [00:18<01:03, 5032.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 92566/414113 [00:18<01:04, 5003.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 22%|██▏       | 93073/414113 [00:18<01:03, 5022.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 93589/414113 [00:19<01:03, 5059.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 94096/414113 [00:19<01:03, 5062.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 94619/414113 [00:19<01:02, 5108.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 95131/414113 [00:19<01:02, 5064.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 95649/414113 [00:19<01:02, 5096.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 96159/414113 [00:19<01:03, 5042.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 96671/414113 [00:19<01:02, 5063.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 23%|██▎       | 97192/414113 [00:19<01:02, 5105.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▎       | 97703/414113 [00:19<01:02, 5086.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▎       | 98212/414113 [00:19<01:02, 5045.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▍       | 98717/414113 [00:20<01:02, 5011.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▍       | 99224/414113 [00:20<01:02, 5027.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▍       | 99727/414113 [00:20<01:02, 4998.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▍       | 100227/414113 [00:20<01:03, 4969.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▍       | 100725/414113 [00:20<01:03, 4950.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 24%|██▍       | 101221/414113 [00:20<01:03, 4909.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▍       | 101720/414113 [00:20<01:03, 4932.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▍       | 102214/414113 [00:20<01:05, 4745.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▍       | 102735/414113 [00:20<01:03, 4873.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▍       | 103241/414113 [00:20<01:03, 4927.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▌       | 103752/414113 [00:21<01:02, 4980.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▌       | 104260/414113 [00:21<01:01, 5009.23it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▌       | 104762/414113 [00:21<01:02, 4956.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▌       | 105265/414113 [00:21<01:02, 4974.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▌       | 105763/414113 [00:21<01:02, 4913.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▌       | 106261/414113 [00:21<01:02, 4930.93it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▌       | 106763/414113 [00:21<01:02, 4956.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▌       | 107259/414113 [00:21<01:03, 4833.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▌       | 107758/414113 [00:21<01:02, 4878.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▌       | 108267/414113 [00:21<01:01, 4937.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▋       | 108781/414113 [00:22<01:01, 4994.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 26%|██▋       | 109282/414113 [00:22<01:01, 4958.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 109800/414113 [00:22<01:00, 5021.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 110305/414113 [00:22<01:00, 5027.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 110809/414113 [00:22<01:00, 5024.78it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 111312/414113 [00:22<01:00, 5015.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 111814/414113 [00:22<01:00, 4985.67it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 112327/414113 [00:22<01:00, 5025.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 112830/414113 [00:22<01:00, 4981.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 113343/414113 [00:23<00:59, 5023.31it/s]\u001b[A\u001b[A\n",
      "\n",
      " 27%|██▋       | 113855/414113 [00:23<00:59, 5048.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 114363/414113 [00:23<00:59, 5055.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 114875/414113 [00:23<00:58, 5072.14it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 115383/414113 [00:23<00:59, 4988.09it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 115889/414113 [00:23<00:59, 5006.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 116397/414113 [00:23<00:59, 5026.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 116900/414113 [00:23<00:59, 5018.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 117403/414113 [00:23<00:59, 5013.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 28%|██▊       | 117905/414113 [00:23<01:06, 4455.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▊       | 118407/414113 [00:24<01:04, 4610.25it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▊       | 118910/414113 [00:24<01:02, 4726.51it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 119425/414113 [00:24<01:00, 4844.83it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 119939/414113 [00:24<00:59, 4929.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 120463/414113 [00:24<00:58, 5017.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 120969/414113 [00:24<00:58, 5026.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 121491/414113 [00:24<00:57, 5081.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 122002/414113 [00:24<00:57, 5043.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|██▉       | 122508/414113 [00:24<00:58, 5022.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|██▉       | 123012/414113 [00:24<00:58, 4944.92it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|██▉       | 123522/414113 [00:25<00:58, 4989.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|██▉       | 124033/414113 [00:25<00:57, 5024.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|███       | 124548/414113 [00:25<00:57, 5059.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|███       | 125069/414113 [00:25<00:56, 5101.83it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|███       | 125584/414113 [00:25<00:56, 5116.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 30%|███       | 126096/414113 [00:25<00:56, 5065.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███       | 126613/414113 [00:25<00:56, 5094.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███       | 127127/414113 [00:25<00:56, 5107.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███       | 127638/414113 [00:25<00:56, 5050.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███       | 128144/414113 [00:25<00:56, 5052.36it/s]\u001b[A\u001b[A\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 31%|███       | 128650/414113 [00:26<00:57, 4991.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███       | 129150/414113 [00:26<00:57, 4965.81it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███▏      | 129678/414113 [00:26<00:56, 5055.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 31%|███▏      | 130185/414113 [00:26<00:56, 5003.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 130692/414113 [00:26<00:56, 5019.69it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 131195/414113 [00:26<00:56, 5022.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 131698/414113 [00:26<00:57, 4948.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 132206/414113 [00:26<00:56, 4984.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 132705/414113 [00:26<00:56, 4949.25it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 133201/414113 [00:26<00:57, 4925.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 133700/414113 [00:27<00:56, 4943.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 32%|███▏      | 134195/414113 [00:27<00:56, 4931.24it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 134720/414113 [00:27<00:55, 5021.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 135230/414113 [00:27<00:55, 5042.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 135739/414113 [00:27<00:55, 5053.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 136254/414113 [00:27<00:54, 5079.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 136766/414113 [00:27<00:54, 5091.25it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 137277/414113 [00:27<00:54, 5095.29it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 137787/414113 [00:27<00:55, 5001.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 33%|███▎      | 138299/414113 [00:27<00:54, 5036.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▎      | 138811/414113 [00:28<00:54, 5058.61it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▎      | 139318/414113 [00:28<00:54, 5045.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 139828/414113 [00:28<00:54, 5060.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 140335/414113 [00:28<00:55, 4946.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 140831/414113 [00:28<00:57, 4733.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 141323/414113 [00:28<00:56, 4786.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 141825/414113 [00:28<00:56, 4853.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 142348/414113 [00:28<00:54, 4958.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 34%|███▍      | 142846/414113 [00:28<00:54, 4957.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▍      | 143370/414113 [00:29<00:53, 5036.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▍      | 143875/414113 [00:29<00:53, 5022.52it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▍      | 144400/414113 [00:29<00:53, 5087.23it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▍      | 144913/414113 [00:29<00:52, 5098.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▌      | 145449/414113 [00:29<00:51, 5174.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▌      | 145968/414113 [00:29<00:52, 5145.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▌      | 146484/414113 [00:29<00:52, 5109.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▌      | 146996/414113 [00:29<00:53, 5015.53it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▌      | 147499/414113 [00:29<00:53, 4999.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▌      | 148000/414113 [00:29<00:54, 4925.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▌      | 148512/414113 [00:30<00:53, 4982.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▌      | 149015/414113 [00:30<00:53, 4995.50it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▌      | 149515/414113 [00:30<00:53, 4982.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▌      | 150014/414113 [00:30<00:53, 4957.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▋      | 150510/414113 [00:30<00:53, 4948.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 36%|███▋      | 151006/414113 [00:30<00:53, 4940.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 151505/414113 [00:30<00:53, 4954.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 152014/414113 [00:30<00:52, 4993.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 152514/414113 [00:30<00:52, 4988.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 153013/414113 [00:30<00:52, 4985.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 153520/414113 [00:31<00:52, 5009.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 154034/414113 [00:31<00:51, 5045.83it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 154539/414113 [00:31<00:51, 4996.35it/s]\u001b[A\u001b[A\n",
      "\n",
      " 37%|███▋      | 155039/414113 [00:31<00:51, 4985.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 155538/414113 [00:31<00:52, 4971.09it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 156036/414113 [00:31<00:55, 4655.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 156542/414113 [00:31<00:54, 4769.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 157076/414113 [00:31<00:52, 4925.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 157582/414113 [00:31<00:51, 4963.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 158089/414113 [00:31<00:51, 4994.22it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 158600/414113 [00:32<00:50, 5027.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 38%|███▊      | 159113/414113 [00:32<00:50, 5055.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▊      | 159644/414113 [00:32<00:49, 5128.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▊      | 160159/414113 [00:32<00:49, 5134.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▉      | 160674/414113 [00:32<00:49, 5071.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▉      | 161184/414113 [00:32<00:49, 5077.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▉      | 161693/414113 [00:32<00:50, 5040.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▉      | 162210/414113 [00:32<00:49, 5075.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▉      | 162727/414113 [00:32<00:49, 5101.13it/s]\u001b[A\u001b[A\n",
      "\n",
      " 39%|███▉      | 163238/414113 [00:32<00:50, 4959.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|███▉      | 163745/414113 [00:33<00:50, 4991.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|███▉      | 164255/414113 [00:33<00:49, 5022.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|███▉      | 164758/414113 [00:33<00:49, 5004.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|███▉      | 165259/414113 [00:33<00:50, 4950.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|████      | 165755/414113 [00:33<00:50, 4912.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|████      | 166247/414113 [00:33<00:50, 4876.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|████      | 166736/414113 [00:33<00:50, 4880.50it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|████      | 167255/414113 [00:33<00:49, 4967.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 167758/414113 [00:33<00:49, 4984.14it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 168267/414113 [00:34<00:49, 5013.92it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 168769/414113 [00:34<00:49, 5006.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 169279/414113 [00:34<00:48, 5031.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 169783/414113 [00:34<00:48, 5001.22it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 170292/414113 [00:34<00:48, 5027.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████      | 170804/414113 [00:34<00:48, 5054.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████▏     | 171310/414113 [00:34<00:48, 4978.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 41%|████▏     | 171809/414113 [00:34<00:48, 4965.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 172316/414113 [00:34<00:48, 4995.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 172824/414113 [00:34<00:48, 5017.67it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 173326/414113 [00:35<00:48, 4973.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 173824/414113 [00:35<01:31, 2629.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 174302/414113 [00:35<01:18, 3039.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 174801/414113 [00:35<01:09, 3441.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 175287/414113 [00:35<01:03, 3771.86it/s]\u001b[A\u001b[A\n",
      "\n",
      " 42%|████▏     | 175798/414113 [00:35<00:58, 4092.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 176279/414113 [00:35<00:55, 4281.72it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 176788/414113 [00:36<00:52, 4494.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 177301/414113 [00:36<00:50, 4666.51it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 177794/414113 [00:36<00:53, 4449.67it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 178294/414113 [00:36<00:51, 4599.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 178798/414113 [00:36<00:49, 4720.81it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 179299/414113 [00:36<00:48, 4802.78it/s]\u001b[A\u001b[A\n",
      "\n",
      " 43%|████▎     | 179788/414113 [00:36<00:48, 4815.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▎     | 180276/414113 [00:36<00:48, 4800.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▎     | 180780/414113 [00:36<00:47, 4869.99it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 181271/414113 [00:36<00:47, 4880.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 181764/414113 [00:37<00:47, 4895.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 182257/414113 [00:37<00:47, 4904.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 182753/414113 [00:37<00:47, 4919.14it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 183246/414113 [00:37<00:47, 4895.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 183742/414113 [00:37<00:46, 4914.25it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 184234/414113 [00:37<00:46, 4903.35it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▍     | 184725/414113 [00:37<00:47, 4871.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▍     | 185214/414113 [00:37<00:46, 4874.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▍     | 185732/414113 [00:37<00:46, 4959.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▍     | 186229/414113 [00:37<00:46, 4948.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▌     | 186746/414113 [00:38<00:45, 5012.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▌     | 187258/414113 [00:38<00:44, 5043.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▌     | 187763/414113 [00:38<00:45, 5020.53it/s]\u001b[A\u001b[A\n",
      "\n",
      " 45%|████▌     | 188276/414113 [00:38<00:44, 5052.58it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▌     | 188782/414113 [00:38<00:44, 5049.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▌     | 189289/414113 [00:38<00:44, 5054.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▌     | 189795/414113 [00:38<00:44, 4990.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▌     | 190296/414113 [00:38<00:44, 4994.46it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▌     | 190800/414113 [00:38<00:44, 5006.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▌     | 191301/414113 [00:38<00:44, 5006.23it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▋     | 191802/414113 [00:39<00:44, 5000.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 46%|████▋     | 192306/414113 [00:39<00:44, 5010.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 192808/414113 [00:39<00:44, 4982.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 193307/414113 [00:39<00:44, 4952.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 193819/414113 [00:39<00:44, 4998.84it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 194320/414113 [00:39<00:44, 4975.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 194818/414113 [00:39<00:44, 4968.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 195334/414113 [00:39<00:43, 5023.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 195866/414113 [00:39<00:42, 5107.57it/s]\u001b[A\u001b[A\n",
      "\n",
      " 47%|████▋     | 196378/414113 [00:39<00:43, 5039.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 196884/414113 [00:40<00:43, 5042.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 197399/414113 [00:40<00:42, 5073.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 197912/414113 [00:40<00:42, 5089.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 198422/414113 [00:40<00:42, 5054.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 198928/414113 [00:40<00:43, 5003.86it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 199429/414113 [00:40<00:42, 5001.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 199930/414113 [00:40<00:42, 4987.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 48%|████▊     | 200452/414113 [00:40<00:42, 5052.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▊     | 200965/414113 [00:40<00:42, 5074.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▊     | 201481/414113 [00:40<00:41, 5097.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 201991/414113 [00:41<00:41, 5089.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 202501/414113 [00:41<00:41, 5073.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 203009/414113 [00:41<00:41, 5057.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 203521/414113 [00:41<00:41, 5075.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 204029/414113 [00:41<00:42, 4998.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 204530/414113 [00:41<00:42, 4985.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|████▉     | 205031/414113 [00:41<00:41, 4990.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|████▉     | 205536/414113 [00:41<00:41, 5005.67it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|████▉     | 206042/414113 [00:41<00:41, 5018.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|████▉     | 206545/414113 [00:41<00:41, 5021.61it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|████▉     | 207048/414113 [00:42<00:41, 5011.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|█████     | 207555/414113 [00:42<00:41, 5028.16it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|█████     | 208060/414113 [00:42<00:40, 5033.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|█████     | 208579/414113 [00:42<00:40, 5077.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 50%|█████     | 209087/414113 [00:42<00:40, 5077.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████     | 209606/414113 [00:42<00:40, 5110.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████     | 210128/414113 [00:42<00:39, 5142.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████     | 210643/414113 [00:42<00:39, 5087.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████     | 211152/414113 [00:42<00:40, 4952.50it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████     | 211649/414113 [00:43<00:40, 4954.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████     | 212146/414113 [00:43<00:40, 4956.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████▏    | 212668/414113 [00:43<00:40, 5030.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 51%|█████▏    | 213178/414113 [00:43<00:39, 5050.93it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 213691/414113 [00:43<00:39, 5073.00it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 214199/414113 [00:43<00:39, 5054.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 214723/414113 [00:43<00:39, 5106.72it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 215234/414113 [00:43<00:39, 5050.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 215754/414113 [00:43<00:38, 5093.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 216264/414113 [00:43<00:38, 5082.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 216774/414113 [00:44<00:38, 5086.00it/s]\u001b[A\u001b[A\n",
      "\n",
      " 52%|█████▏    | 217284/414113 [00:44<00:38, 5086.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 217793/414113 [00:44<00:38, 5075.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 218317/414113 [00:44<00:38, 5122.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 218854/414113 [00:44<00:37, 5191.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 219374/414113 [00:44<00:37, 5142.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 219910/414113 [00:44<00:37, 5203.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 220431/414113 [00:44<00:37, 5194.53it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 220959/414113 [00:44<00:37, 5217.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 53%|█████▎    | 221482/414113 [00:44<00:37, 5174.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▎    | 222002/414113 [00:45<00:37, 5180.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▎    | 222521/414113 [00:45<00:37, 5147.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 223036/414113 [00:45<00:37, 5108.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 223555/414113 [00:45<00:37, 5130.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 224070/414113 [00:45<00:37, 5134.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 224584/414113 [00:45<00:37, 5071.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 225098/414113 [00:45<00:37, 5090.52it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 225617/414113 [00:45<00:36, 5116.07it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▍    | 226133/414113 [00:45<00:36, 5128.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▍    | 226678/414113 [00:45<00:35, 5218.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▍    | 227201/414113 [00:46<00:36, 5140.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▍    | 227716/414113 [00:46<00:36, 5137.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▌    | 228231/414113 [00:46<00:36, 5119.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▌    | 228744/414113 [00:46<00:36, 5116.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▌    | 229256/414113 [00:46<00:36, 5074.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 55%|█████▌    | 229764/414113 [00:46<00:36, 5007.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▌    | 230266/414113 [00:46<00:36, 4979.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▌    | 230765/414113 [00:46<00:36, 4966.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▌    | 231262/414113 [00:46<00:36, 4948.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▌    | 231758/414113 [00:46<00:37, 4904.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▌    | 232249/414113 [00:47<00:37, 4847.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▌    | 232743/414113 [00:47<00:37, 4873.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▋    | 233244/414113 [00:47<00:36, 4912.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 56%|█████▋    | 233738/414113 [00:47<00:36, 4918.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 234254/414113 [00:47<00:36, 4988.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 234754/414113 [00:47<00:36, 4965.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 235251/414113 [00:47<00:36, 4947.92it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 235754/414113 [00:47<00:35, 4970.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 236252/414113 [00:47<00:35, 4971.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 236750/414113 [00:47<00:36, 4893.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 237240/414113 [00:48<00:36, 4892.46it/s]\u001b[A\u001b[A\n",
      "\n",
      " 57%|█████▋    | 237730/414113 [00:48<00:36, 4893.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 238224/414113 [00:48<00:35, 4905.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 238734/414113 [00:48<00:35, 4960.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 239241/414113 [00:48<00:35, 4991.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 239741/414113 [00:48<00:35, 4966.47it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 240238/414113 [00:48<00:35, 4954.66it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 240740/414113 [00:48<00:34, 4973.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 241241/414113 [00:48<00:34, 4983.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 241740/414113 [00:48<00:34, 4971.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 58%|█████▊    | 242243/414113 [00:49<00:34, 4987.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▊    | 242746/414113 [00:49<00:34, 4999.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▊    | 243246/414113 [00:49<00:34, 4977.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 243755/414113 [00:49<00:34, 5008.38it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 244265/414113 [00:49<00:33, 5033.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 244769/414113 [00:49<00:33, 5006.84it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 245270/414113 [00:49<00:33, 4990.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 245783/414113 [00:49<00:33, 5028.07it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 246306/414113 [00:49<00:32, 5086.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|█████▉    | 246815/414113 [00:49<00:33, 5048.72it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|█████▉    | 247335/414113 [00:50<00:32, 5090.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|█████▉    | 247856/414113 [00:50<00:32, 5124.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|█████▉    | 248369/414113 [00:50<00:32, 5055.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|██████    | 248875/414113 [00:50<00:33, 4999.52it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|██████    | 249395/414113 [00:50<00:32, 5056.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|██████    | 249902/414113 [00:50<00:32, 5025.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 60%|██████    | 250405/414113 [00:50<00:32, 5023.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████    | 250908/414113 [00:50<00:32, 5008.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████    | 251417/414113 [00:50<00:32, 5031.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████    | 251921/414113 [00:50<00:32, 5029.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████    | 252425/414113 [00:51<00:32, 5017.24it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████    | 252927/414113 [00:51<00:32, 4981.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████    | 253426/414113 [00:51<00:32, 4936.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████▏   | 253920/414113 [00:51<00:32, 4933.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 61%|██████▏   | 254414/414113 [00:51<00:32, 4919.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 254921/414113 [00:51<00:32, 4962.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 255440/414113 [00:51<00:31, 5026.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 255951/414113 [00:51<00:31, 5049.63it/s]\u001b[A\u001b[A\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 62%|██████▏   | 256457/414113 [00:51<00:31, 5045.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 256962/414113 [00:51<00:31, 5035.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 257494/414113 [00:52<00:30, 5116.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 258007/414113 [00:52<00:30, 5101.16it/s]\u001b[A\u001b[A\n",
      "\n",
      " 62%|██████▏   | 258518/414113 [00:52<00:32, 4741.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 259016/414113 [00:52<00:32, 4810.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 259529/414113 [00:52<00:31, 4900.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 260023/414113 [00:52<00:31, 4906.72it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 260527/414113 [00:52<00:31, 4944.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 261024/414113 [00:52<00:30, 4945.00it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 261526/414113 [00:52<00:30, 4965.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 262024/414113 [00:53<00:34, 4462.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 63%|██████▎   | 262544/414113 [00:53<00:32, 4659.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▎   | 263040/414113 [00:53<00:31, 4743.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▎   | 263550/414113 [00:53<00:31, 4844.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 264040/414113 [00:53<00:31, 4806.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 264530/414113 [00:53<00:30, 4832.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 265021/414113 [00:53<00:30, 4853.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 265509/414113 [00:53<00:30, 4830.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 266013/414113 [00:53<00:30, 4891.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 266528/414113 [00:53<00:29, 4964.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 267030/414113 [00:54<00:29, 4980.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▍   | 267544/414113 [00:54<00:29, 5024.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▍   | 268048/414113 [00:54<00:29, 5025.86it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▍   | 268552/414113 [00:54<00:29, 5014.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▍   | 269054/414113 [00:54<00:29, 4972.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▌   | 269552/414113 [00:54<00:29, 4917.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▌   | 270045/414113 [00:54<00:29, 4898.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▌   | 270536/414113 [00:54<00:29, 4892.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 65%|██████▌   | 271026/414113 [00:54<00:29, 4859.23it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▌   | 271513/414113 [00:54<00:29, 4833.14it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▌   | 272005/414113 [00:55<00:29, 4857.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▌   | 272491/414113 [00:55<00:29, 4849.09it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▌   | 272998/414113 [00:55<00:28, 4910.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▌   | 273490/414113 [00:55<00:30, 4651.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▌   | 273975/414113 [00:55<00:29, 4708.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▋   | 274473/414113 [00:55<00:29, 4786.02it/s]\u001b[A\u001b[A\n",
      "\n",
      " 66%|██████▋   | 274974/414113 [00:55<00:28, 4849.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 275477/414113 [00:55<00:28, 4900.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 275971/414113 [00:55<00:28, 4911.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 276468/414113 [00:56<00:27, 4928.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 276962/414113 [00:56<00:27, 4927.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 277456/414113 [00:56<00:28, 4876.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 277951/414113 [00:56<00:27, 4897.00it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 278450/414113 [00:56<00:27, 4923.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 278943/414113 [00:56<00:27, 4884.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 67%|██████▋   | 279433/414113 [00:56<00:27, 4887.92it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 279929/414113 [00:56<00:27, 4908.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 280420/414113 [00:56<00:27, 4900.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 280911/414113 [00:56<00:27, 4894.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 281401/414113 [00:57<00:27, 4875.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 281889/414113 [00:57<00:27, 4830.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 282373/414113 [00:57<00:27, 4832.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 282876/414113 [00:57<00:26, 4889.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 68%|██████▊   | 283375/414113 [00:57<00:26, 4916.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▊   | 283878/414113 [00:57<00:26, 4948.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▊   | 284374/414113 [00:57<00:26, 4941.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 284878/414113 [00:57<00:26, 4970.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 285391/414113 [00:57<00:25, 5014.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 285893/414113 [00:57<00:26, 4891.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 286383/414113 [00:58<00:26, 4888.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 286883/414113 [00:58<00:25, 4919.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 287376/414113 [00:58<00:25, 4901.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|██████▉   | 287895/414113 [00:58<00:25, 4984.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|██████▉   | 288396/414113 [00:58<00:25, 4991.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|██████▉   | 288899/414113 [00:58<00:25, 5001.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|██████▉   | 289404/414113 [00:58<00:24, 5013.93it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|███████   | 289906/414113 [00:58<00:24, 5006.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|███████   | 290407/414113 [00:58<00:25, 4864.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|███████   | 290910/414113 [00:58<00:25, 4911.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|███████   | 291409/414113 [00:59<00:24, 4934.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 70%|███████   | 291910/414113 [00:59<00:24, 4956.46it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████   | 292408/414113 [00:59<00:24, 4961.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████   | 292921/414113 [00:59<00:24, 5009.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████   | 293426/414113 [00:59<00:24, 5021.25it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████   | 293929/414113 [00:59<00:23, 5007.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████   | 294442/414113 [00:59<00:23, 5042.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████   | 294947/414113 [00:59<00:23, 5005.78it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████▏  | 295449/414113 [00:59<00:23, 5007.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 71%|███████▏  | 295950/414113 [00:59<00:23, 4996.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 296457/414113 [01:00<00:23, 5016.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 296966/414113 [01:00<00:23, 5037.83it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 297470/414113 [01:00<00:23, 4999.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 297971/414113 [01:00<00:23, 4994.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 298473/414113 [01:00<00:23, 5000.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 298974/414113 [01:00<00:23, 4961.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 299482/414113 [01:00<00:22, 4993.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 72%|███████▏  | 300000/414113 [01:00<00:22, 5047.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 300505/414113 [01:00<00:22, 4941.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 301013/414113 [01:00<00:22, 4980.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 301516/414113 [01:01<00:22, 4992.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 302027/414113 [01:01<00:22, 5026.69it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 302540/414113 [01:01<00:22, 5055.13it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 303046/414113 [01:01<00:22, 5010.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 303548/414113 [01:01<00:22, 4997.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 73%|███████▎  | 304048/414113 [01:01<00:22, 4986.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▎  | 304558/414113 [01:01<00:21, 5018.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▎  | 305061/414113 [01:01<00:21, 4958.22it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▍  | 305558/414113 [01:01<00:21, 4952.00it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▍  | 306054/414113 [01:01<00:21, 4927.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▍  | 306547/414113 [01:02<00:21, 4900.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▍  | 307038/414113 [01:02<00:21, 4898.93it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▍  | 307548/414113 [01:02<00:21, 4957.50it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▍  | 308058/414113 [01:02<00:21, 4999.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▍  | 308559/414113 [01:02<00:21, 4998.35it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▍  | 309060/414113 [01:02<00:21, 4980.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▍  | 309563/414113 [01:02<00:20, 4993.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▍  | 310069/414113 [01:02<00:20, 5010.72it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▍  | 310571/414113 [01:02<00:21, 4783.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▌  | 311055/414113 [01:02<00:21, 4799.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▌  | 311557/414113 [01:03<00:21, 4862.92it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▌  | 312071/414113 [01:03<00:20, 4941.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 75%|███████▌  | 312593/414113 [01:03<00:20, 5021.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▌  | 313102/414113 [01:03<00:20, 5039.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▌  | 313607/414113 [01:03<00:20, 5024.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▌  | 314111/414113 [01:03<00:20, 4998.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▌  | 314612/414113 [01:03<00:20, 4964.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▌  | 315126/414113 [01:03<00:19, 5014.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▌  | 315628/414113 [01:03<00:19, 4994.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▋  | 316135/414113 [01:04<00:19, 5015.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 76%|███████▋  | 316640/414113 [01:04<00:19, 5025.05it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 317173/414113 [01:04<00:18, 5111.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 317701/414113 [01:04<00:18, 5160.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 318223/414113 [01:04<00:18, 5175.99it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 318741/414113 [01:04<00:18, 5151.67it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 319257/414113 [01:04<00:18, 5148.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 319773/414113 [01:04<00:18, 5124.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 320286/414113 [01:04<00:18, 5096.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 77%|███████▋  | 320798/414113 [01:04<00:18, 5101.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 321309/414113 [01:05<00:18, 5035.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 321821/414113 [01:05<00:18, 5057.52it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 322327/414113 [01:05<00:18, 5042.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 322837/414113 [01:05<00:18, 5057.09it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 323362/414113 [01:05<00:17, 5111.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 323874/414113 [01:05<00:17, 5042.82it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 324379/414113 [01:05<00:36, 2464.78it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 324904/414113 [01:06<00:30, 2930.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▊  | 325408/414113 [01:06<00:26, 3350.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▊  | 325925/414113 [01:06<00:23, 3745.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▉  | 326456/414113 [01:06<00:21, 4106.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▉  | 326960/414113 [01:06<00:20, 4347.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▉  | 327462/414113 [01:06<00:19, 4528.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▉  | 327963/414113 [01:06<00:18, 4661.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▉  | 328472/414113 [01:06<00:17, 4780.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 79%|███████▉  | 328973/414113 [01:06<00:17, 4806.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|███████▉  | 329470/414113 [01:06<00:17, 4843.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|███████▉  | 329981/414113 [01:07<00:17, 4920.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|███████▉  | 330482/414113 [01:07<00:17, 4916.99it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|███████▉  | 330980/414113 [01:07<00:17, 4841.81it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|████████  | 331469/414113 [01:07<00:17, 4824.50it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|████████  | 331955/414113 [01:07<00:17, 4813.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|████████  | 332444/414113 [01:07<00:16, 4834.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 80%|████████  | 332929/414113 [01:07<00:16, 4828.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████  | 333433/414113 [01:07<00:16, 4887.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████  | 333939/414113 [01:07<00:16, 4936.02it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████  | 334452/414113 [01:07<00:15, 4989.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████  | 334954/414113 [01:08<00:15, 4995.57it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████  | 335455/414113 [01:08<00:15, 4984.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████  | 335967/414113 [01:08<00:15, 5022.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████▏ | 336470/414113 [01:08<00:15, 5024.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████▏ | 336973/414113 [01:08<00:15, 5004.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 81%|████████▏ | 337474/414113 [01:08<00:15, 4959.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 337982/414113 [01:08<00:15, 4991.78it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 338482/414113 [01:08<00:15, 4990.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 338982/414113 [01:08<00:15, 4947.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 339487/414113 [01:08<00:14, 4977.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 339985/414113 [01:09<00:14, 4963.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 340484/414113 [01:09<00:14, 4970.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 340987/414113 [01:09<00:14, 4988.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 341492/414113 [01:09<00:14, 5006.09it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 341993/414113 [01:09<00:14, 4990.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 342493/414113 [01:09<00:14, 4990.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 343005/414113 [01:09<00:14, 5025.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 343515/414113 [01:09<00:13, 5046.69it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 344025/414113 [01:09<00:13, 5061.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 344532/414113 [01:10<00:13, 5022.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 345036/414113 [01:10<00:13, 5026.87it/s]\u001b[A\u001b[A\n",
      "\n",
      " 83%|████████▎ | 345539/414113 [01:10<00:13, 5025.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▎ | 346068/414113 [01:10<00:13, 5100.29it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▎ | 346579/414113 [01:10<00:13, 5046.13it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▍ | 347084/414113 [01:10<00:13, 5002.52it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▍ | 347585/414113 [01:10<00:13, 4977.36it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▍ | 348083/414113 [01:10<00:13, 4935.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▍ | 348577/414113 [01:10<00:13, 4913.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▍ | 349069/414113 [01:10<00:13, 4902.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 84%|████████▍ | 349560/414113 [01:11<00:13, 4903.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▍ | 350055/414113 [01:11<00:13, 4916.68it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▍ | 350553/414113 [01:11<00:12, 4933.83it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▍ | 351047/414113 [01:11<00:12, 4922.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▍ | 351540/414113 [01:11<00:12, 4850.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▌ | 352027/414113 [01:11<00:12, 4854.93it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▌ | 352534/414113 [01:11<00:12, 4915.16it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▌ | 353026/414113 [01:11<00:12, 4885.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▌ | 353538/414113 [01:11<00:12, 4952.10it/s]\u001b[A\u001b[A\n",
      "\n",
      " 85%|████████▌ | 354040/414113 [01:11<00:12, 4970.81it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 354557/414113 [01:12<00:11, 5025.53it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 355076/414113 [01:12<00:11, 5073.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 355584/414113 [01:12<00:11, 5060.03it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 356091/414113 [01:12<00:11, 5060.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 356598/414113 [01:12<00:11, 5061.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 357105/414113 [01:12<00:11, 5030.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▋ | 357609/414113 [01:12<00:11, 4990.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▋ | 358122/414113 [01:12<00:11, 5030.85it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 358626/414113 [01:12<00:11, 4999.32it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 359127/414113 [01:12<00:10, 4999.95it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 359632/414113 [01:13<00:10, 5012.90it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 360148/414113 [01:13<00:10, 5055.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 360654/414113 [01:13<00:10, 5053.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 361160/414113 [01:13<00:10, 4928.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 361668/414113 [01:13<00:10, 4973.24it/s]\u001b[A\u001b[A\n",
      "\n",
      " 87%|████████▋ | 362166/414113 [01:13<00:10, 4959.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 362663/414113 [01:13<00:10, 4932.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 363159/414113 [01:13<00:10, 4938.48it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 363659/414113 [01:13<00:10, 4954.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 364168/414113 [01:13<00:10, 4993.99it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 364676/414113 [01:14<00:09, 5015.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 365184/414113 [01:14<00:09, 5032.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 365691/414113 [01:14<00:09, 5042.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 88%|████████▊ | 366196/414113 [01:14<00:09, 5038.91it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▊ | 366707/414113 [01:14<00:09, 5057.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▊ | 367213/414113 [01:14<00:09, 5019.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 367716/414113 [01:14<00:09, 5005.07it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 368217/414113 [01:14<00:09, 4948.15it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 368717/414113 [01:14<00:09, 4961.25it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 369244/414113 [01:14<00:08, 5049.23it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 369750/414113 [01:15<00:08, 5036.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 370254/414113 [01:15<00:08, 5031.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|████████▉ | 370758/414113 [01:15<00:08, 5014.58it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|████████▉ | 371267/414113 [01:15<00:08, 5035.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|████████▉ | 371771/414113 [01:15<00:08, 4992.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|████████▉ | 372271/414113 [01:15<00:08, 4890.34it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|█████████ | 372761/414113 [01:15<00:08, 4873.18it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|█████████ | 373253/414113 [01:15<00:08, 4885.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|█████████ | 373763/414113 [01:15<00:08, 4946.99it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|█████████ | 374259/414113 [01:15<00:08, 4923.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 90%|█████████ | 374768/414113 [01:16<00:07, 4969.75it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████ | 375266/414113 [01:16<00:07, 4962.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████ | 375784/414113 [01:16<00:07, 5024.46it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████ | 376288/414113 [01:16<00:07, 5027.27it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████ | 376791/414113 [01:16<00:07, 4943.37it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████ | 377286/414113 [01:16<00:07, 4910.01it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████ | 377778/414113 [01:16<00:07, 4883.58it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████▏| 378295/414113 [01:16<00:07, 4965.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 91%|█████████▏| 378802/414113 [01:16<00:07, 4995.81it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 379303/414113 [01:16<00:07, 4967.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 379801/414113 [01:17<00:06, 4946.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 380299/414113 [01:17<00:06, 4954.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 380795/414113 [01:17<00:06, 4953.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 381296/414113 [01:17<00:06, 4969.70it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 381795/414113 [01:17<00:06, 4974.45it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 382293/414113 [01:17<00:06, 4943.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 92%|█████████▏| 382788/414113 [01:17<00:06, 4920.43it/s]\u001b[A\u001b[A\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      " 93%|█████████▎| 383308/414113 [01:17<00:06, 4998.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 383813/414113 [01:17<00:06, 5011.73it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 384315/414113 [01:17<00:06, 4930.64it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 384809/414113 [01:18<00:05, 4932.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 385306/414113 [01:18<00:05, 4940.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 385826/414113 [01:18<00:05, 5015.76it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 386340/414113 [01:18<00:05, 5050.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 386846/414113 [01:18<00:05, 5039.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▎| 387353/414113 [01:18<00:05, 5048.13it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▎| 387859/414113 [01:18<00:05, 5020.11it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▍| 388362/414113 [01:18<00:05, 5020.79it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▍| 388879/414113 [01:18<00:04, 5063.14it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▍| 389408/414113 [01:18<00:04, 5128.13it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▍| 389922/414113 [01:19<00:04, 5085.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▍| 390434/414113 [01:19<00:04, 5094.60it/s]\u001b[A\u001b[A\n",
      "\n",
      " 94%|█████████▍| 390944/414113 [01:19<00:04, 5065.19it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▍| 391454/414113 [01:19<00:04, 5073.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▍| 391962/414113 [01:19<00:04, 4993.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▍| 392468/414113 [01:19<00:04, 5011.96it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▍| 392970/414113 [01:19<00:04, 4938.77it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▌| 393476/414113 [01:19<00:04, 4973.58it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▌| 393981/414113 [01:19<00:04, 4995.55it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▌| 394481/414113 [01:20<00:03, 4974.61it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▌| 394979/414113 [01:20<00:03, 4975.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 95%|█████████▌| 395477/414113 [01:20<00:03, 4951.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▌| 395973/414113 [01:20<00:03, 4909.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▌| 396473/414113 [01:20<00:03, 4936.12it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▌| 396970/414113 [01:20<00:03, 4943.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▌| 397465/414113 [01:20<00:03, 4933.89it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▌| 397959/414113 [01:20<00:03, 4929.72it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▌| 398475/414113 [01:20<00:03, 4994.59it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▋| 398987/414113 [01:20<00:03, 5031.35it/s]\u001b[A\u001b[A\n",
      "\n",
      " 96%|█████████▋| 399491/414113 [01:21<00:02, 5002.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 399992/414113 [01:21<00:02, 4968.04it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 400490/414113 [01:21<00:02, 4963.39it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 401006/414113 [01:21<00:02, 5019.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 401509/414113 [01:21<00:02, 5002.86it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 402010/414113 [01:21<00:02, 4980.02it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 402527/414113 [01:21<00:02, 5035.43it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 403036/414113 [01:21<00:02, 5050.54it/s]\u001b[A\u001b[A\n",
      "\n",
      " 97%|█████████▋| 403542/414113 [01:21<00:02, 5023.08it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 404045/414113 [01:21<00:02, 4622.31it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 404562/414113 [01:22<00:02, 4772.69it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 405045/414113 [01:22<00:01, 4750.98it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 405555/414113 [01:22<00:01, 4849.94it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 406064/414113 [01:22<00:01, 4916.88it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 406559/414113 [01:22<00:01, 4857.09it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 407047/414113 [01:22<00:01, 4856.51it/s]\u001b[A\u001b[A\n",
      "\n",
      " 98%|█████████▊| 407542/414113 [01:22<00:01, 4881.44it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▊| 408047/414113 [01:22<00:01, 4929.42it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▊| 408548/414113 [01:22<00:01, 4952.17it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▉| 409044/414113 [01:22<00:01, 4935.81it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▉| 409554/414113 [01:23<00:00, 4982.28it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▉| 410058/414113 [01:23<00:00, 4996.65it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▉| 410576/414113 [01:23<00:00, 5047.50it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▉| 411082/414113 [01:23<00:00, 5048.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 99%|█████████▉| 411596/414113 [01:23<00:00, 5075.05it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|█████████▉| 412104/414113 [01:23<00:00, 5033.53it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|█████████▉| 412619/414113 [01:23<00:00, 5066.39it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|█████████▉| 413126/414113 [01:23<00:00, 5050.70it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|█████████▉| 413648/414113 [01:23<00:00, 5098.59it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|██████████| 414113/414113 [01:23<00:00, 4931.73it/s]\u001b[A\u001b[ADownloading: \"https://download.pytorch.org/models/resnet50-19c8e357.pth\" to /root/.torch/models/resnet50-19c8e357.pth\n",
      "\n",
      "\n",
      "  0%|          | 0/102502400 [00:00<?, ?it/s]\u001b[A\u001b[A\n",
      "\n",
      "  4%|▍         | 4079616/102502400 [00:00<00:02, 40660673.02it/s]\u001b[A\u001b[A\n",
      "\n",
      " 11%|█         | 10911744/102502400 [00:00<00:01, 46279904.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 15%|█▍        | 15286272/102502400 [00:00<00:01, 45480659.33it/s]\u001b[A\u001b[A\n",
      "\n",
      " 19%|█▉        | 19595264/102502400 [00:00<00:01, 44638332.62it/s]\u001b[A\u001b[A\n",
      "\n",
      " 25%|██▍       | 25493504/102502400 [00:00<00:01, 48127721.06it/s]\u001b[A\u001b[A\n",
      "\n",
      " 29%|██▉       | 29802496/102502400 [00:00<00:01, 44876585.71it/s]\u001b[A\u001b[A\n",
      "\n",
      " 35%|███▌      | 36339712/102502400 [00:00<00:01, 49534946.20it/s]\u001b[A\u001b[A\n",
      "\n",
      " 40%|████      | 41230336/102502400 [00:00<00:01, 41803120.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 44%|████▍     | 45572096/102502400 [00:01<00:01, 34587555.40it/s]\u001b[A\u001b[A\n",
      "\n",
      " 49%|████▉     | 50708480/102502400 [00:01<00:01, 38344058.63it/s]\u001b[A\u001b[A\n",
      "\n",
      " 54%|█████▍    | 55427072/102502400 [00:01<00:01, 40625969.56it/s]\u001b[A\u001b[A\n",
      "\n",
      " 59%|█████▉    | 60555264/102502400 [00:01<00:00, 43057122.21it/s]\u001b[A\u001b[A\n",
      "\n",
      " 64%|██████▍   | 65863680/102502400 [00:01<00:00, 45617128.26it/s]\u001b[A\u001b[A\n",
      "\n",
      " 69%|██████▉   | 70639616/102502400 [00:01<00:00, 45840459.74it/s]\u001b[A\u001b[A\n",
      "\n",
      " 74%|███████▎  | 75374592/102502400 [00:01<00:00, 43080159.80it/s]\u001b[A\u001b[A\n",
      "\n",
      " 78%|███████▊  | 79822848/102502400 [00:01<00:00, 36393174.30it/s]\u001b[A\u001b[A\n",
      "\n",
      " 82%|████████▏ | 83820544/102502400 [00:01<00:00, 37397836.97it/s]\u001b[A\u001b[A\n",
      "\n",
      " 86%|████████▌ | 87769088/102502400 [00:02<00:00, 37901357.49it/s]\u001b[A\u001b[A\n",
      "\n",
      " 89%|████████▉ | 91701248/102502400 [00:02<00:00, 37961665.41it/s]\u001b[A\u001b[A\n",
      "\n",
      " 93%|█████████▎| 95641600/102502400 [00:02<00:00, 38362983.30it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|█████████▉| 102105088/102502400 [00:02<00:00, 43690302.39it/s]\u001b[A\u001b[A\n",
      "\n",
      "100%|██████████| 102502400/102502400 [00:02<00:00, 43482676.75it/s]\u001b[A\u001b[A"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Done (t=0.84s)\n",
      "creating index...\n",
      "index created!\n",
      "Obtaining caption lengths...\n"
     ]
    }
   ],
   "source": [
    "import torch\n",
    "import torch.nn as nn\n",
    "from torchvision import transforms\n",
    "import sys\n",
    "sys.path.append('/opt/cocoapi/PythonAPI')\n",
    "from pycocotools.coco import COCO\n",
    "from data_loader import get_loader\n",
    "from model import EncoderCNN, DecoderRNN\n",
    "import math\n",
    "\n",
    "import nltk\n",
    "nltk.download('punkt')\n",
    "\n",
    "## TODO #1: Select appropriate values for the Python variables below.\n",
    "batch_size = 64          # batch size\n",
    "vocab_threshold = 5        # minimum word count threshold\n",
    "vocab_from_file = True    # if True, load existing vocab file\n",
    "embed_size = 512           # dimensionality of image and word embeddings\n",
    "hidden_size = 512          # number of features in hidden state of the RNN decoder\n",
    "num_epochs = 3             # number of training epochs\n",
    "save_every = 1             # determines frequency of saving model weights\n",
    "print_every = 100          # determines window for printing average loss\n",
    "log_file = 'training_log.txt'       # name of file with saved training loss and perplexity\n",
    "\n",
    "# (Optional) TODO #2: Amend the image transform below.\n",
    "transform_train = transforms.Compose([ \n",
    "    transforms.Resize(256),                          # smaller edge of image resized to 256\n",
    "    transforms.RandomCrop(224),                      # get 224x224 crop from random location\n",
    "    transforms.RandomHorizontalFlip(),               # horizontally flip image with probability=0.5\n",
    "    transforms.ToTensor(),                           # convert the PIL Image to a tensor\n",
    "    transforms.Normalize((0.485, 0.456, 0.406),      # normalize image for pre-trained model\n",
    "                         (0.229, 0.224, 0.225))])\n",
    "\n",
    "# Build data loader.\n",
    "data_loader = get_loader(transform=transform_train,\n",
    "                         mode='train',\n",
    "                         batch_size=batch_size,\n",
    "                         vocab_threshold=vocab_threshold,\n",
    "                         vocab_from_file=vocab_from_file)\n",
    "\n",
    "# The size of the vocabulary.\n",
    "vocab_size = len(data_loader.dataset.vocab)\n",
    "\n",
    "# Initialize the encoder and decoder. \n",
    "encoder = EncoderCNN(embed_size)\n",
    "decoder = DecoderRNN(embed_size, hidden_size, vocab_size)\n",
    "\n",
    "# Move models to GPU if CUDA is available. \n",
    "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
    "encoder.to(device)\n",
    "decoder.to(device)\n",
    "\n",
    "# Define the loss function. \n",
    "criterion = nn.CrossEntropyLoss().cuda() if torch.cuda.is_available() else nn.CrossEntropyLoss()\n",
    "\n",
    "# TODO #3: Specify the learnable parameters of the model.\n",
    "params = list(decoder.parameters()) + list(encoder.embed.parameters()) + list(encoder.bn.parameters())\n",
    "\n",
    "# TODO #4: Define the optimizer.\n",
    "optimizer = torch.optim.Adam(params, lr=0.001, betas=(0.9, 0.999))\n",
    "\n",
    "# Set the total number of training steps per epoch.\n",
    "total_step = math.ceil(len(data_loader.dataset.caption_lengths) / data_loader.batch_sampler.batch_size)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id='step2'></a>\n",
    "## Step 2: Train your Model\n",
    "\n",
    "Once you have executed the code cell in **Step 1**, the training procedure below should run without issue.  \n",
    "\n",
    "It is completely fine to leave the code cell below as-is without modifications to train your model.  However, if you would like to modify the code used to train the model below, you must ensure that your changes are easily parsed by your reviewer.  In other words, make sure to provide appropriate comments to describe how your code works!  \n",
    "\n",
    "You may find it useful to load saved weights to resume training.  In that case, note the names of the files containing the encoder and decoder weights that you'd like to load (`encoder_file` and `decoder_file`).  Then you can load the weights by using the lines below:\n",
    "\n",
    "```python\n",
    "# Load pre-trained weights before resuming training.\n",
    "encoder.load_state_dict(torch.load(os.path.join('./models', encoder_file)))\n",
    "decoder.load_state_dict(torch.load(os.path.join('./models', decoder_file)))\n",
    "```\n",
    "\n",
    "While trying out parameters, make sure to take extensive notes and record the settings that you used in your various training runs.  In particular, you don't want to encounter a situation where you've trained a model for several hours but can't remember what settings you used :).\n",
    "\n",
    "### A Note on Tuning Hyperparameters\n",
    "\n",
    "To figure out how well your model is doing, you can look at how the training loss and perplexity evolve during training - and for the purposes of this project, you are encouraged to amend the hyperparameters based on this information.  \n",
    "\n",
    "However, this will not tell you if your model is overfitting to the training data, and, unfortunately, overfitting is a problem that is commonly encountered when training image captioning models.  \n",
    "\n",
    "For this project, you need not worry about overfitting. **This project does not have strict requirements regarding the performance of your model**, and you just need to demonstrate that your model has learned **_something_** when you generate captions on the test data.  For now, we strongly encourage you to train your model for the suggested 3 epochs without worrying about performance; then, you should immediately transition to the next notebook in the sequence (**3_Inference.ipynb**) to see how your model performs on the test data.  If your model needs to be changed, you can come back to this notebook, amend hyperparameters (if necessary), and re-train the model.\n",
    "\n",
    "That said, if you would like to go above and beyond in this project, you can read about some approaches to minimizing overfitting in section 4.3.1 of [this paper](http://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=7505636).  In the next (optional) step of this notebook, we provide some guidance for assessing the performance on the validation dataset."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch [1/3], Step [100/6471], Loss: 4.1844, Perplexity: 65.6542\n",
      "Epoch [1/3], Step [200/6471], Loss: 4.2649, Perplexity: 71.16093\n",
      "Epoch [1/3], Step [300/6471], Loss: 3.5860, Perplexity: 36.09047\n",
      "Epoch [1/3], Step [400/6471], Loss: 3.9075, Perplexity: 49.77303\n",
      "Epoch [1/3], Step [500/6471], Loss: 3.6566, Perplexity: 38.7310\n",
      "Epoch [1/3], Step [600/6471], Loss: 3.2409, Perplexity: 25.5573\n",
      "Epoch [1/3], Step [700/6471], Loss: 3.3017, Perplexity: 27.1586\n",
      "Epoch [1/3], Step [800/6471], Loss: 3.0736, Perplexity: 21.6198\n",
      "Epoch [1/3], Step [900/6471], Loss: 2.9552, Perplexity: 19.2059\n",
      "Epoch [1/3], Step [1000/6471], Loss: 2.9027, Perplexity: 18.2232\n",
      "Epoch [1/3], Step [1100/6471], Loss: 3.2778, Perplexity: 26.5174\n",
      "Epoch [1/3], Step [1200/6471], Loss: 2.8794, Perplexity: 17.8044\n",
      "Epoch [1/3], Step [1300/6471], Loss: 2.8414, Perplexity: 17.1394\n",
      "Epoch [1/3], Step [1400/6471], Loss: 2.5894, Perplexity: 13.3211\n",
      "Epoch [1/3], Step [1500/6471], Loss: 2.7909, Perplexity: 16.2965\n",
      "Epoch [1/3], Step [1600/6471], Loss: 3.7082, Perplexity: 40.7813\n",
      "Epoch [1/3], Step [1700/6471], Loss: 2.8120, Perplexity: 16.6438\n",
      "Epoch [1/3], Step [1800/6471], Loss: 2.7903, Perplexity: 16.2855\n",
      "Epoch [1/3], Step [1900/6471], Loss: 2.7612, Perplexity: 15.81895\n",
      "Epoch [1/3], Step [2000/6471], Loss: 2.8567, Perplexity: 17.4043\n",
      "Epoch [1/3], Step [2100/6471], Loss: 2.6269, Perplexity: 13.8307\n",
      "Epoch [1/3], Step [2200/6471], Loss: 2.7168, Perplexity: 15.1322\n",
      "Epoch [1/3], Step [2300/6471], Loss: 2.6166, Perplexity: 13.6891\n",
      "Epoch [1/3], Step [2400/6471], Loss: 2.5548, Perplexity: 12.8684\n",
      "Epoch [1/3], Step [2500/6471], Loss: 2.5843, Perplexity: 13.2546\n",
      "Epoch [1/3], Step [2600/6471], Loss: 2.4816, Perplexity: 11.9598\n",
      "Epoch [1/3], Step [2700/6471], Loss: 2.4278, Perplexity: 11.33410\n",
      "Epoch [1/3], Step [2800/6471], Loss: 2.4377, Perplexity: 11.4466\n",
      "Epoch [1/3], Step [2900/6471], Loss: 2.9469, Perplexity: 19.0463\n",
      "Epoch [1/3], Step [3000/6471], Loss: 2.7298, Perplexity: 15.3300\n",
      "Epoch [1/3], Step [3100/6471], Loss: 2.5949, Perplexity: 13.3951\n",
      "Epoch [1/3], Step [3200/6471], Loss: 2.6361, Perplexity: 13.9588\n",
      "Epoch [1/3], Step [3300/6471], Loss: 2.4489, Perplexity: 11.5760\n",
      "Epoch [1/3], Step [3400/6471], Loss: 2.7842, Perplexity: 16.1864\n",
      "Epoch [1/3], Step [3500/6471], Loss: 2.5621, Perplexity: 12.9625\n",
      "Epoch [1/3], Step [3600/6471], Loss: 2.2876, Perplexity: 9.85164\n",
      "Epoch [1/3], Step [3700/6471], Loss: 2.3134, Perplexity: 10.1083\n",
      "Epoch [1/3], Step [3800/6471], Loss: 2.3392, Perplexity: 10.3733\n",
      "Epoch [1/3], Step [3900/6471], Loss: 2.3393, Perplexity: 10.3739\n",
      "Epoch [1/3], Step [4000/6471], Loss: 2.2408, Perplexity: 9.40119\n",
      "Epoch [1/3], Step [4100/6471], Loss: 2.3249, Perplexity: 10.22557\n",
      "Epoch [1/3], Step [4200/6471], Loss: 2.4999, Perplexity: 12.1808\n",
      "Epoch [1/3], Step [4300/6471], Loss: 2.2856, Perplexity: 9.83142\n",
      "Epoch [1/3], Step [4400/6471], Loss: 2.3593, Perplexity: 10.5832\n",
      "Epoch [1/3], Step [4500/6471], Loss: 2.5456, Perplexity: 12.7504\n",
      "Epoch [1/3], Step [4600/6471], Loss: 2.2935, Perplexity: 9.90988\n",
      "Epoch [1/3], Step [4700/6471], Loss: 2.0758, Perplexity: 7.97081\n",
      "Epoch [1/3], Step [4800/6471], Loss: 2.4558, Perplexity: 11.6563\n",
      "Epoch [1/3], Step [4900/6471], Loss: 2.5632, Perplexity: 12.9771\n",
      "Epoch [1/3], Step [5000/6471], Loss: 2.1856, Perplexity: 8.89602\n",
      "Epoch [1/3], Step [5100/6471], Loss: 2.4951, Perplexity: 12.1227\n",
      "Epoch [1/3], Step [5200/6471], Loss: 2.2349, Perplexity: 9.34524\n",
      "Epoch [1/3], Step [5300/6471], Loss: 2.6734, Perplexity: 14.4884\n",
      "Epoch [1/3], Step [5400/6471], Loss: 2.1718, Perplexity: 8.77398\n",
      "Epoch [1/3], Step [5500/6471], Loss: 2.3019, Perplexity: 9.99303\n",
      "Epoch [1/3], Step [5600/6471], Loss: 2.2698, Perplexity: 9.67778\n",
      "Epoch [1/3], Step [5700/6471], Loss: 2.2340, Perplexity: 9.33742\n",
      "Epoch [1/3], Step [5800/6471], Loss: 2.2160, Perplexity: 9.17028\n",
      "Epoch [1/3], Step [5900/6471], Loss: 2.4736, Perplexity: 11.8646\n",
      "Epoch [1/3], Step [6000/6471], Loss: 2.3363, Perplexity: 10.3432\n",
      "Epoch [1/3], Step [6100/6471], Loss: 2.1358, Perplexity: 8.46409\n",
      "Epoch [1/3], Step [6200/6471], Loss: 1.9729, Perplexity: 7.19187\n",
      "Epoch [1/3], Step [6300/6471], Loss: 2.2576, Perplexity: 9.56054\n",
      "Epoch [1/3], Step [6400/6471], Loss: 2.1870, Perplexity: 8.90881\n",
      "Epoch [2/3], Step [100/6471], Loss: 2.0558, Perplexity: 7.813481\n",
      "Epoch [2/3], Step [200/6471], Loss: 2.3665, Perplexity: 10.6601\n",
      "Epoch [2/3], Step [300/6471], Loss: 2.2187, Perplexity: 9.19529\n",
      "Epoch [2/3], Step [400/6471], Loss: 2.4821, Perplexity: 11.9669\n",
      "Epoch [2/3], Step [500/6471], Loss: 2.4252, Perplexity: 11.3043\n",
      "Epoch [2/3], Step [600/6471], Loss: 2.4026, Perplexity: 11.0521\n",
      "Epoch [2/3], Step [700/6471], Loss: 2.8947, Perplexity: 18.0778\n",
      "Epoch [2/3], Step [800/6471], Loss: 2.1932, Perplexity: 8.96365\n",
      "Epoch [2/3], Step [900/6471], Loss: 2.2992, Perplexity: 9.96673\n",
      "Epoch [2/3], Step [1000/6471], Loss: 2.1579, Perplexity: 8.6533\n",
      "Epoch [2/3], Step [1100/6471], Loss: 2.1514, Perplexity: 8.59712\n",
      "Epoch [2/3], Step [1200/6471], Loss: 2.1828, Perplexity: 8.87136\n",
      "Epoch [2/3], Step [1300/6471], Loss: 2.4462, Perplexity: 11.5439\n",
      "Epoch [2/3], Step [1400/6471], Loss: 2.3126, Perplexity: 10.1002\n",
      "Epoch [2/3], Step [1500/6471], Loss: 1.9559, Perplexity: 7.07037\n",
      "Epoch [2/3], Step [1600/6471], Loss: 2.1704, Perplexity: 8.76182\n",
      "Epoch [2/3], Step [1700/6471], Loss: 2.2612, Perplexity: 9.59484\n",
      "Epoch [2/3], Step [1800/6471], Loss: 2.7861, Perplexity: 16.2171\n",
      "Epoch [2/3], Step [1900/6471], Loss: 1.9379, Perplexity: 6.94423\n",
      "Epoch [2/3], Step [2000/6471], Loss: 2.2281, Perplexity: 9.28228\n",
      "Epoch [2/3], Step [2100/6471], Loss: 2.4162, Perplexity: 11.2030\n",
      "Epoch [2/3], Step [2200/6471], Loss: 2.1948, Perplexity: 8.97832\n",
      "Epoch [2/3], Step [2300/6471], Loss: 2.2224, Perplexity: 9.22996\n",
      "Epoch [2/3], Step [2352/6471], Loss: 2.3143, Perplexity: 10.1177"
     ]
    }
   ],
   "source": [
    "import torch.utils.data as data\n",
    "import numpy as np\n",
    "import os\n",
    "import requests\n",
    "import time\n",
    "\n",
    "# Open the training log file.\n",
    "f = open(log_file, 'w')\n",
    "\n",
    "old_time = time.time()\n",
    "response = requests.request(\"GET\", \n",
    "                            \"http://metadata.google.internal/computeMetadata/v1/instance/attributes/keep_alive_token\", \n",
    "                            headers={\"Metadata-Flavor\":\"Google\"})\n",
    "\n",
    "for epoch in range(1, num_epochs+1):\n",
    "    \n",
    "    for i_step in range(1, total_step+1):\n",
    "        \n",
    "        if time.time() - old_time > 60:\n",
    "            old_time = time.time()\n",
    "            requests.request(\"POST\", \n",
    "                             \"https://nebula.udacity.com/api/v1/remote/keep-alive\", \n",
    "                             headers={'Authorization': \"STAR \" + response.text})\n",
    "        \n",
    "        # Randomly sample a caption length, and sample indices with that length.\n",
    "        indices = data_loader.dataset.get_train_indices()\n",
    "        # Create and assign a batch sampler to retrieve a batch with the sampled indices.\n",
    "        new_sampler = data.sampler.SubsetRandomSampler(indices=indices)\n",
    "        data_loader.batch_sampler.sampler = new_sampler\n",
    "        \n",
    "        # Obtain the batch.\n",
    "        images, captions = next(iter(data_loader))\n",
    "\n",
    "        # Move batch of images and captions to GPU if CUDA is available.\n",
    "        images = images.to(device)\n",
    "        captions = captions.to(device)\n",
    "        \n",
    "        # Zero the gradients.\n",
    "        decoder.zero_grad()\n",
    "        encoder.zero_grad()\n",
    "        \n",
    "        # Pass the inputs through the CNN-RNN model.\n",
    "        features = encoder(images)\n",
    "        outputs = decoder(features, captions)\n",
    "        \n",
    "        # Calculate the batch loss.\n",
    "        loss = criterion(outputs.view(-1, vocab_size), captions.view(-1))\n",
    "        \n",
    "        # Backward pass.\n",
    "        loss.backward()\n",
    "        \n",
    "        # Update the parameters in the optimizer.\n",
    "        optimizer.step()\n",
    "            \n",
    "        # Get training statistics.\n",
    "        stats = 'Epoch [%d/%d], Step [%d/%d], Loss: %.4f, Perplexity: %5.4f' % (epoch, num_epochs, i_step, total_step, loss.item(), np.exp(loss.item()))\n",
    "        \n",
    "        # Print training statistics (on same line).\n",
    "        print('\\r' + stats, end=\"\")\n",
    "        sys.stdout.flush()\n",
    "        \n",
    "        # Print training statistics to file.\n",
    "        f.write(stats + '\\n')\n",
    "        f.flush()\n",
    "        \n",
    "        # Print training statistics (on different line).\n",
    "        if i_step % print_every == 0:\n",
    "            print('\\r' + stats)\n",
    "            \n",
    "    # Save the weights.\n",
    "    if epoch % save_every == 0:\n",
    "        torch.save(decoder.state_dict(), os.path.join('./models', 'decoder-%d.pkl' % epoch))\n",
    "        torch.save(encoder.state_dict(), os.path.join('./models', 'encoder-%d.pkl' % epoch))\n",
    "\n",
    "# Close the training log file.\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id='step3'></a>\n",
    "## Step 3: (Optional) Validate your Model\n",
    "\n",
    "To assess potential overfitting, one approach is to assess performance on a validation set.  If you decide to do this **optional** task, you are required to first complete all of the steps in the next notebook in the sequence (**3_Inference.ipynb**); as part of that notebook, you will write and test code (specifically, the `sample` method in the `DecoderRNN` class) that uses your RNN decoder to generate captions.  That code will prove incredibly useful here. \n",
    "\n",
    "If you decide to validate your model, please do not edit the data loader in **data_loader.py**.  Instead, create a new file named **data_loader_val.py** containing the code for obtaining the data loader for the validation data.  You can access:\n",
    "- the validation images at filepath `'/opt/cocoapi/images/train2014/'`, and\n",
    "- the validation image caption annotation file at filepath `'/opt/cocoapi/annotations/captions_val2014.json'`.\n",
    "\n",
    "The suggested approach to validating your model involves creating a json file such as [this one](https://github.com/cocodataset/cocoapi/blob/master/results/captions_val2014_fakecap_results.json) containing your model's predicted captions for the validation images.  Then, you can write your own script or use one that you [find online](https://github.com/tylin/coco-caption) to calculate the BLEU score of your model.  You can read more about the BLEU score, along with other evaluation metrics (such as TEOR and Cider) in section 4.1 of [this paper](https://arxiv.org/pdf/1411.4555.pdf).  For more information about how to use the annotation file, check out the [website](http://cocodataset.org/#download) for the COCO dataset."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# (Optional) TODO: Validate your model."
   ]
  }
 ],
 "metadata": {
  "anaconda-cloud": {},
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}