Python-Indexer错误：索引1超出大小为1的轴0的界限

我对机器学习相当陌生，我正在尝试做图像处理。基本上预测模型的准确性，该模型将确定彩色图像上的人脸是否戴着面具。数据分为包含两个文件夹（带\u掩码，不带\u掩码）的测试目录和包含两个文件夹（带\u掩码，不带\u掩码）的列车目录。我在网上看不同的例子来实现我的目标。我正在使用GoogleColab，所以我将代码划分为不同的代码单元，并在运行

model.fit({'input': train_x}, {'targets': train_y}, n_epoch=10, validation_set=({'input': test_x}, {'targets': test_y}),snapshot_step = 100, show_metric=True,run_id = MODEL_NAME )
model.save(MODEL_NAME)

我遇到了错误

索引器：索引1超出了大小为1的轴0的界限，并且它似乎被捕获在一个无限循环中。下面我包括了完整的代码以及完整的错误

代码：

错误：

Run id: withMaskvsnoMask-0.001-6conv-basic.model
Log directory: /tmp/tflearn_logs/
---------------------------------
Training samples: 3
Validation samples: 3
--
Exception in thread Thread-79:
Traceback (most recent call last):
  File "/usr/lib/python3.6/threading.py", line 916, in _bootstrap_inner
    self.run()
  File "/usr/lib/python3.6/threading.py", line 864, in run
    self._target(*self._args, **self._kwargs)
  File "/usr/local/lib/python3.6/dist-packages/tflearn/data_flow.py", line 187, in fill_feed_dict_queue
    data = self.retrieve_data(batch_ids)
  File "/usr/local/lib/python3.6/dist-packages/tflearn/data_flow.py", line 222, in retrieve_data
    utils.slice_array(self.feed_dict[key], batch_ids)
  File "/usr/local/lib/python3.6/dist-packages/tflearn/utils.py", line 204, in slice_array
    return X[start]
IndexError: index 1 is out of bounds for axis 0 with size 1

---------------------------------------------------------------------------
KeyboardInterrupt                         Traceback (most recent call last)
<ipython-input-61-10937b30276a> in <module>()
      1 #Oviya
----> 2 model.fit({'input': train_x}, {'targets': train_y}, n_epoch=10, validation_set=({'input': test_x}, {'targets': test_y}),snapshot_step = 100, show_metric=True,run_id = MODEL_NAME )
      3 model.save(MODEL_NAME)
      4 

5 frames
/usr/lib/python3.6/threading.py in wait(self, timeout)
    293         try:    # restore state no matter what (e.g., KeyboardInterrupt)
    294             if timeout is None:
--> 295                 waiter.acquire()
    296                 gotit = True
    297             else:

KeyboardInterrupt: 

运行id:withMaskvsnoMask-0.001-6conv-basic.model
日志目录：/tmp/tflearn\u日志/
---------------------------------
培训样本：3
验证样本：3
--
线程-79中的异常：
回溯（最近一次呼叫最后一次）：
文件“/usr/lib/python3.6/threading.py”，第916行，在内部引导中
self.run（）
文件“/usr/lib/python3.6/threading.py”，第864行，运行中
自我目标（*自我参数，**自我参数）
文件“/usr/local/lib/python3.6/dist packages/tflearn/data\u flow.py”，第187行，在fill\u feed\u dict\u队列中
数据=自身。检索数据（批处理ID）
文件“/usr/local/lib/python3.6/dist packages/tflearn/data\u flow.py”，第222行，在retrieve\u data中
utils.slice\u数组（self.feed\u dict[key]，batch\u id）
文件“/usr/local/lib/python3.6/dist-packages/tflearn/utils.py”，第204行，在切片数组中
返回X[开始]
索引器：索引1超出大小为1的轴0的界限
---------------------------------------------------------------------------
键盘中断回溯（最后一次最近调用）
在（）
1#Oviya
---->2.model.fit（{'input'：train_x}，{'targets'：train_y}，n_epoch=10，validation_set=（{'input'：test_x}，{'targets'：test_y}），snapshot_step=100，show_metric=True，run_id=model_NAME）
3模型。保存（模型名称）
4.
5帧
/usr/lib/python3.6/threading.py处于等待状态（self，超时）
293 try:#无论发生什么情况都恢复状态（例如键盘中断）
294如果超时为无：
-->服务生
296 gotit=正确
297其他：
键盘中断：

Run id: withMaskvsnoMask-0.001-6conv-basic.model
Log directory: /tmp/tflearn_logs/
---------------------------------
Training samples: 3
Validation samples: 3
--
Exception in thread Thread-79:
Traceback (most recent call last):
  File "/usr/lib/python3.6/threading.py", line 916, in _bootstrap_inner
    self.run()
  File "/usr/lib/python3.6/threading.py", line 864, in run
    self._target(*self._args, **self._kwargs)
  File "/usr/local/lib/python3.6/dist-packages/tflearn/data_flow.py", line 187, in fill_feed_dict_queue
    data = self.retrieve_data(batch_ids)
  File "/usr/local/lib/python3.6/dist-packages/tflearn/data_flow.py", line 222, in retrieve_data
    utils.slice_array(self.feed_dict[key], batch_ids)
  File "/usr/local/lib/python3.6/dist-packages/tflearn/utils.py", line 204, in slice_array
    return X[start]
IndexError: index 1 is out of bounds for axis 0 with size 1

---------------------------------------------------------------------------
KeyboardInterrupt                         Traceback (most recent call last)
<ipython-input-61-10937b30276a> in <module>()
      1 #Oviya
----> 2 model.fit({'input': train_x}, {'targets': train_y}, n_epoch=10, validation_set=({'input': test_x}, {'targets': test_y}),snapshot_step = 100, show_metric=True,run_id = MODEL_NAME )
      3 model.save(MODEL_NAME)
      4 

5 frames
/usr/lib/python3.6/threading.py in wait(self, timeout)
    293         try:    # restore state no matter what (e.g., KeyboardInterrupt)
    294             if timeout is None:
--> 295                 waiter.acquire()
    296                 gotit = True
    297             else:

KeyboardInterrupt: