import torch
import numpy as np
import os
import json
from EduNLP.Pretrain import train_elmo, ElmoTokenizer
from EduNLP.Vector import ElmoModel, T2V
from EduNLP.I2V import Elmo, get_pretrained_i2v

d:\MySoftwares\Anaconda\envs\data\lib\site-packages\gensim\similarities\__init__.py:15: UserWarning: The gensim.similarities.levenshtein submodule is disabled, because the optional Levenshtein package <https://pypi.org/project/python-Levenshtein/> is unavailable. Install Levenhstein (e.g. `pip install python-Levenshtein`) to suppress this warning.
  warnings.warn(msg)

训练自己的Elmo模型

1. 数据

# 设置你的数据路径和输出路径
BASE_DIR = "../.."

data_dir = f"{BASE_DIR}/static/test_data"
output_dir = f"{BASE_DIR}/examples/test_model/elmo"

def raw_data():
    _data = []
    data_path = os.path.join(data_dir, "OpenLUNA.json")
    with open(data_path, encoding="utf-8") as f:
        for line in f.readlines():
            _data.append(json.loads(line))
    return _data

def stem_data(data):
    _data = []
    tokenizer = ElmoTokenizer()
    for e in data:
        d = tokenizer.tokenize(item=e['stem'], freeze_vocab=False)
        if d is not None:
            _data.append(d)
    assert _data
    return _data

raw_data = raw_data()
train_items = stem_data(raw_data)

2. 训练和评估

# 自定义训练参数
train_params = {
  "emb_dim": 128,
  "hid_dim": 256,
  "batch_size": 4,
  "epochs": 1,
  "lr": 5e-3,
  "device": None
}

train_elmo(train_items, output_dir, **train_params)

RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   13,   15,   16,   17,   18,
           19,   17,   13,   34,  125,   11,   91,  248,   15,   16,   13,   19,
           29,   18,   34,  248,   15,   16,   19,   13,   34,   29,   18,   24,
           93, 1181,  253,  125,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  132,   16,  125,   92,  131,   18,  125,   92,   29,   34,  125,
           92,   20,   10,   96,  584,  125,   92,   17,   34,  125,   92,   21,
           10,   30,  584, 1785,   92,   29,  125,   92,   17,  837,  125,   92,
          131,  168,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  229,  116,   16,   29,   18,   16,   17,   34,  116,   16,   29,
           18,   16,   17,   34,  116,   16,   29,   18,   16,   17,   18,   18,
           18,  129,  593,  995, 1159,  997,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   32,   33,  116,   16,   29,   34,   33,   18,   16,   29,   19,
           33,   18,   10,   35, 1722,   33, 1722,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 125,  257,  170,  257,   23,   23,   22,  162,   22,   29,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61, 1393, 1662,  138,  242, 1393, 1599, 1662, 1515,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   26,  889,   16,  170,   18,  350,
          351,   15,   16,   82,   18,   26,   27,   36,  889,   16,  125,   18,
           36,   10,   17,   26,   27,   36,  889,   16,  170,   18,   36,   10,
           29,   36,  889,   16,  125,   18,   34,   17,  889,   16,  170,   18,
           36,  737,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  519,  520,  149,  523,  878,  879,  359,  880,  352,  523,  804,
          519,  247,  116,   16,  881,  117,   18,   16,   20,   18,  158,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,  230,   91,  115,   13,   34,
          796,   93,   11,   91,  115,  257,   23,   26,   27,   36,  796,   36,
          578,  116,   16,  117,   18,   16,   17,   18,   24,   93,   26,   27,
           13,   10,   19,  116,   16,  117,   18,   16,   21,   18,  113,   91,
           13,   93,  253,   83,   16,   13,   18,   10,  116,   16,  117,   18,
           16,   21,   18,   83,   16,   79,   18,   10,   83,   16,  113,   18,
           91,   83,   16,   13,   18,   93,  797, 1783,   83,   16,  113,   18,
           91,   83,   16,   13,   18,   93,   11,   91,  116,   16,  117,   18,
           16, 1484,   18,   26,  116,   16,   30,  117,   18,   16, 1426,   18,
           24,   93,  251,  115,  168,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 353,   92,  131,  180,   12,  125,   92,  131,   25,  181,  131,  183,
            7,  125,   92,  131,  962,   23,  584,  125,   92,  131,   15,   17,
           34,   17,  125,   92,  131,   10,   21,  353,   92,  131,   34,   20,
          229,   16,  125,   92,  131,   18,  526,  527,   61,  170,   92,  131,
           10,  116,   16,   29,   18,   16,  125,   92,  602,   92,   16,  131,
           34,   29,   18,   18,  229,  180,   16,  170,   92,  131,   18,  181,
          131,  183,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  798,  129,  355, 1415,  436,  130,  610,   54, 1021,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1880, 1881,  276,  102,  173,   65,  273,  178,   11,   12,  160,  161,
           16,  277,   18,   13,   10,   20,  114,  288,  164,   79,   10,  230,
          288,  166,  161,   26,   91,  288,  478,   16,  479,   18,   93,   24,
          167,  107,  502,  273,  178,   11,   12,  160,  161,   16,  277,   18,
           83,   16,   13,   18,   10,   83,   16,  125,   18,   34,   21,   83,
           16,  279,   18,  164,   83,   16,   79,   18,   10,   29,   19,   83,
           16,  279,   18,  166,  161,   26,   27,   91,   83,   16,  279,   18,
          478,   16,  479,   18,   93,   24,  167,  125,  229,   65,  502,  267,
          140,   65,   68,   36,  358,  168, 1856,  228, 1048,  229,  125,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 173,   79,   10,   17,  174,   91,   13,   34,   29,   93,   68,  579,
          176,  178,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  91,  125,   34,   13,   93,   91,   29,   34,   13,   93,   15,   16,
           21,   18,  582,  102,   13, 2040, 2041, 2042,  583, 2005,  125,  179,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 136,  116,   16,   13,   15,   16,   17,   18,   18,   16,  125,   15,
           16,   17,   18,   18,   19,  116,   16,   79,   15,   16,   17,   18,
           18,   16,  170,   15,   16,   17,   18,   18,   10,   29,   27,   91,
          125,  257,   23,   26,  170,  257,   23,   93,  457,  228,   16,   20,
           18,  360,  178,   91,   27,   93,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 125,  136,  116,   16,   13,   15,   16,   17,   18,   18,   16,  125,
           15,   16,   17,   18,   18,   19,   79,   15,   16,   17,   18,   10,
           29,  457,  254,   91,   27,   93,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,   13,   79,   33, 1186,   17,   15,   16,   13,   18,   10,   20,
           15,   16,   79,   18,   10,   30,   15,   16,   33,   18,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   91,   29,   34,   17,   35,   93,   91,  125,   34,   35,   93,
          860,  722,  125,  861,  125,   54,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  136,   65,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           27,   91,  125,  257,   23,   26,  170,  257,   23,   93,  457,  228,
           16,   17,   18,   68,   65,  360,  358,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  116,   16,  521,   13,   18,   16,
           29,   34,  521,   15,   16,   17,   18,   13,   18,  580,  123,    4,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  349,  125,   26,  170,  350,  351,   15,   16,   82,   18,   26,
          169,  102,  170,  352,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  290,  291,    7,  125,   26,  170,   26,  162, 1186,  574,   10,
           29,  241,  116,   16,   29,   18,   16,  125,   18,   34,  116,   16,
           29,   18,   16,  170,   18,   34,  116,   16,   29,   18,   16,  162,
           18,  163,  125,   15,   16,   17,   18,   34,  170,   15,   16,   17,
           18,   34,  162,   15,   16,   17,   18,   91,  125,   34,  170,   93,
           15,   16,   20,   18,   34,   91,  170,   34,  162,   93,   15,   16,
           20,   18,   34,   91,  162,   34,  125,   93,   15,   16,   20,   18,
          165,  354,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1361,  441,  893, 1362, 1363,  710, 1364, 1365,  893,  894,  338, 1366,
          894, 1366, 1362, 1367,   97,  705, 1368,  530,  407, 1369, 1370,   85,
         1371, 1368, 1372, 1366, 1373, 1368, 1374, 1366, 1368,  893,  331, 1375,
           81, 1016, 1193, 1376, 1377, 1376, 1377,  554, 1366, 1362, 1378,  449,
          229,   97,  705, 1368,  893,  828, 1375, 1379,  171,  893,  828, 1375,
          339,  171,  339,  171, 1368,  433,  997,  553,  102,  553, 1382,  556,
          894, 1366, 1362,  440,  450,  503,   15,   16,   17,   18,   10,  116,
           16,  131,   91,  539,   19,  540,   93,   15,   16,   17,   18,   18,
           16,   91,  125,   34,  170,   93,   91,  162,   34,  541,   93,   91,
          125,   34,  162,   93,   91,  170,   34,  541,   93,   18,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  139,  135,   92,   16,   29,   18,   91,   19,   29,
           26,   23,   93,   26,   27,  135,   92,   16,   17,   18,   91,   29,
           26,   23,   93,  135,   92,   16,   17,   18,  107,   65,  361,    9,
          584,   28,  362,   11,   36, 1162,   92,   16,   17,   18,   24,   36,
           10,   17,   11,   36,  135,   92,   16,   17,   18,   28,   24,   36,
           26,   36,  153,   36,   10,   11,   36,  562,   92,   16,   29,   18,
           24,   36,   65,  178,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 363,  151,  226,    9,   26,   28,   26,   65,  125,   26,  170,   26,
          162,   26,   27,    7,  114,   15,   16,   17,   18,   11,   91,  116,
           16,  117,   18,   16,   17,   18,   34,    9,   24,   93,   34,  114,
            9,   10,  116,   16,   30,   18,   16,   21,   18,  229,    9,  170,
           19,  162,   10,  116,   16,  228,   16,   20,   18,   18,   16,   20,
           18,  125,   26,  241,  363,  151,  395,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,   83,   16,  151,   18,  226,    9,   28,   65,  125,  170,  162,
            7,  230,    9,   34,  228,   16,   20,   18,  114,    9,   10,   23,
           26,   27,  125,   10,   17,  228,   16,  133,   18,  164,  170,  164,
          229,  162,   61,   66,  154, 1067,  238,  594,  172,  155,  229,  145,
           83,   16, 2016,   18,   53,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  800,  232, 1516,  519,  520,  519,  158,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  174,   13,   34,  125,   91,
           29,   19,   13,   93,  250,  113,   91,   13,   93,  251, 1316,  113,
           91,   13,   93,  168,  168,  754, 1510,  249,  229,  125,  254,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,  297,  298,  299,  300,  301,  125,   92,   16,   29,   18,   26,
          125,   92,   16,   17,   18,   26,   94,  125,   92,   16,  298,   18,
           26,   61,   29,  163,   35,  163,  302,  163,  278,  163,  298,  278,
           19,  302,   10,   20,  302,   19,   35,   10,   21,  303,  125,   92,
           16,   35,   18,   26,  125,   92,   16,  302,   18,   26,  125,   92,
           16,  278,   18,  304,  305,  278,   19,  302,   10,   21,  302,   19,
           35,   10,   20,   26,  303,  125,   92,   16,   35,   18,   26,  125,
           92,   16,  302,   18,   26,  125,   92,   16,  278,   18,  304,  306,
          298,  299,  304,  305,  304,  306,  307,  308,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  513,   65,   79,   15,   16,   17,   18,   10,   21,   13,  139,
          135,  135,  177,  278,  278,  107,  277,   65,  515,    9,   28,  362,
           36,  153,   36,   10,  134,  229,  277,  178, 1645,   68,    9,   28,
           65,  458,  357,  104,  178,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  246,  142,  822,  151,  102,  153,   10,  154,   10,   17,  228,
           16,   17,   18,   83,   16,  266,   18,   10,   83,   16,  268,   18,
           10,   83,   16,  608,   18,   10,   83,   16,  155,   18,   10,   21,
           62,  155,  466,  241,  605,  815,  242,  151,  567,  464,  558,  154,
          607,  464,  822,  266,  822,   65,  596,  820,  229,  608,  242, 2136,
          719,  720,  610,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  116,   16,  248,   15,   16,   13,
           18,   19,  248,   15,   16,   19,   13,   18,   18,   16,   13,   15,
           16,   17,   18,   18,  797,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  464,   91,   13,   92,   23,  584,   79,   92,   23,   93,  136,
           65,  137,  116,   16,   13,   15,   17,   18,   16,   17,   18,   19,
           79,   15,   17,   10,   29,  238,  135,   92,   29,  135,   92,   17,
           65,  258,  259,  138,  139, 1735,   16, 1736,   18,   16,  716,   92,
           29,   18,  263, 1735,   16, 1736,   18,   16,  716,   92,   17,   18,
         1737,   23,   79,   92,   23,  254,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  11,   91,   29,   34,  116,   16,   29,   18,   16,   13,   15,   16,
           17,   18,   18,   24,   93,   91,   29,   34,   13,   93,   15,   16,
          105,   18,  582,  102,   13,   15,   16,   17,   18,  583,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301,   83,   16,  353,   18,   92,   16,  131,   18,  180,   11,   12,
          125,   92,   16,  131,   18,   24,   25,  181,  131,  183,   83,   16,
          353,   18,   92,   16,  131,   18,   10,   17,  125,   92,   16,  131,
           18,   34,   29,   83,   16,  353,   18,   92,   16,  105,   18,   10,
          179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   11,   12,   13,   36,   13,   15,   16,   17,
           18,   19,   13,   19,   17,  257,   23,   24,   25,   11,  118,   92,
           16,  573,   18,    9,   24,  167,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 417,  379,    9,   28,   65,  822, 1775,   79,  949,  464,  465,  362,
           36,  470,   36,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 521,  288,   10,  116,   16,   29,   18,   16,   20,   18,  114,   17,
          288,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  231,  129,  355,  125,   10,   21,  170,   10,  105,  130,  131,
           10,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  145,   83,   16,  956,   18,  957,  669,  239,   83,   16,  958,
           18,   10,  819,   15,   16,   82,   18,  167,  234,   26,  116,   16,
           29,   18,   16,   17,   18,   83,   16,  854,   18,  523,  959,  241,
          107,  153,  150,   62,  357,   68,   65,   66,  150,   62,    9,   28,
           65,   66,  960,  961,  241,  153,  269,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  7, 125, 294,  11,  91,  23,  26, 116,  16, 117,  18,  16,  17,  18,
          24,  93,  17, 230,  17, 384,  10, 114,  17, 384,  34,  29,  26, 230,
         384,  10,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [112, 113,  91,  13,  93,  10,  17, 230,  13,  19, 230,  17,  13, 118,
          23,  26,  17, 117, 119, 253, 307,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7,   9,  26,  28, 255, 256, 137, 116,  16,  13,  15,  16,  17,  18,
          18,  16, 125,  15,  16,  17,  18,  18,  34,  79,  15,  16,  17,  18,
          10,  29,  91, 125, 257,  29,  93, 258, 259, 233, 260, 256, 233,  26,
         261,  16, 262,  18, 263, 261,  16, 264,  18,  10, 134,  26, 142, 107,
          13,  10, 105, 265, 266, 256, 267,  65, 268, 256, 267,  66, 229, 256,
         178, 241, 107, 269, 270,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [271, 272, 273, 178, 274, 275, 276, 102, 173,  65, 273, 178,  11,  12,
         160, 161,  16, 277,  18,  13,  10,  17,  19, 279,  19, 279,  15,  16,
          17,  18, 164,  79,  10,  17,  19,  20, 279,  34, 279,  15,  16,  17,
          18, 166, 161,  26,  91, 279, 478,  16, 755,  18, 279, 572,  29,  24,
         167,  65, 356, 515,   9,  28, 362, 229,  36, 153,  36, 140, 141, 280,
          13, 281, 282, 283, 284, 285, 272, 229, 107, 153, 286, 178,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1092, 1093, 1094, 1095, 1096,  203,  500, 1097, 1098,   90,   81, 1099,
         1095, 1096,  203,  500, 1097, 1098,   90, 1100, 1101, 1102,  912, 1095,
         1096,  203,  500, 1097, 1098,   90,  942, 1101, 1103,  219, 1092, 1095,
         1096, 1097, 1098,  346, 1104,  171,   15,   20,   71,  219, 1092, 1095,
         1096, 1050, 1105, 1106, 1050, 1107,  500,  423,  486,  102,   90, 1108,
           90,  546,  466, 1109,  548,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,   13,   19,  116,   16,   29,   18,
           16,   20,   18,  230,   17,   13,   34,  885,   16,  125,  886,   18,
           13,   91,   19,  887,   26,   34,  887,   93,  251,  888,  125,  254,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  125,   15,   16,   17,   18,   18,   34,  116,   16,   79,   15,
           16,   17,   18,   18,   16,   21,   18,   10,   29,  139,   65,  457,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1398, 1399, 1400, 1401,  980, 1402, 1403,  662,  980, 1404, 1405, 1406,
         1407,  528, 1408, 1405, 1059,  530,  405,  216,  528, 1408, 1398, 1401,
          528,  682,  201, 1409, 1408, 1401,  528,  682,  426, 1409, 1408, 1398,
         1408, 1401,  528,  682,  351, 1409, 1410, 1408, 1398,  528, 1275, 1410,
          528, 1411,   60,  430,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 800,  242,  801,  802,  803,  523,  451,  519,  519,  520,  102,  691,
          692,  798,  804,  519,  158,  182,   34,   85,  117,  451,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 397, 1050,  338, 1051, 1052, 1051, 1053, 1054,  324, 1055, 1056, 1057,
          397, 1051, 1053, 1054, 1058, 1059,  217,  397,  338, 1051, 1052, 1051,
         1053, 1054, 1060, 1061,  121, 1062,  102, 1063,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93,   10,   36,   13,   19,  125,   36,   13,
           34,   36,   13,   19,   17,   36,   91,   13,   19,  125,   93,  125,
           10,   29,  249,  229,  290,  113,   91,   13,   93,   22,   23,  293,
           13,  294,   91,   19,  887,   26,   29,   93,  249,  113,   91,   13,
           93,   22,   23,  229,  125,  254,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 518,  519,  520,  519,  158,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   10,   12,   23,   26,   17,   25,   26,   27,   28,
           10,   12,   19,   17,   26,   19,   29,   26,   23,   26,   29,   26,
           17,   25,    9,   31,   28,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   52,   63,  102,  256,  260, 1088, 1009,  690,  455,  560, 1465,
           66, 1569, 1570,  172,  855, 1571,  135,  241,   28,   65,  260,  135,
          960,  961,  153,  256, 1088,  466,  229,  811, 1572,   53,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 501,  368,  488,  489,  490,  491,  492,  493,  494,  495,   90,  284,
          397,  309,  310,  496,  497,  498,  433,  502,   91,  279,   93,  279,
           81,  500,  488,  489,  502,   91,  279,   93,   10,  116,   16,  503,
           18,   16,   29,   34,  248,   15,   16,   19,   20,   13,   91,  279,
           19,   30,   93,   18,   18,  503,  497,  498,  433,  502,   11,   91,
          279,   15,   16,  506,   18,   24,   93,   10,  347,  503,  249,  507,
          508,  509,  311,  279,   15,   16,  506,   18,  510,   91,  174,  511,
          452,   20,   93,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  136,   65,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           27,   91,  125,  257,   23,   26,  170,  257,   23,   93,  175,  360,
          178,   79,   10,  116,   16,  228,   16,   30,   18,   18,   16,   17,
           18,   13,  255,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          298,   18,   34,  116,   16,   79,   15,   16,   17,   18,   18,   16,
           20,   18,   10,   29,  289,  139,   65,  178,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  125,   10,  126,   92,   16,   17,   18,  841,   26,  170,   10,
           17,   15,   16,  841,   18,   26,  162,   10,  841,   15,   16,  901,
           18,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 513, 2129, 2183,  373,  139,  255,  116,   16,   13,   15,   16,   17,
           18,   18,   16,   20,  373,   18,   34,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  373,   18,   10,   29,  139,  373,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 212,  209, 1543, 1576, 1577,  612,  775, 1578, 1579, 1580, 1581, 1582,
         1578, 1541, 1583,  102, 1409, 1584, 1409,  212,  209, 1543, 1582,  209,
         1543, 1582, 1576,  212, 1582, 1585,  212, 1541, 1582, 1586,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  125,   10,   17,   15,   16,  116,   16,   21,   18,   16,   20,
           18,   18,   26,  170,   10,   20,   15,   16,  116,   16,   17,   18,
           16,   20,   18,   18,   26,  162,   10,  214,   15,   16,  116,   16,
           29,   18,   16,   20,   18,   18,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,   79,   10,   17,   13,   15,   16,   17,   18,   19,  248,   15,
           16,   36,   13,   36,   18,  822,  797,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   49,   47,  142,  822,   63,  102,  266,  815,   59,   63,  594,
         1599,  154,  153,  594,  155,  266,  154,  464, 1089,  594,  238,  467,
         2006,  465,  608,  466,  241,  470, 1599,  242,  243,  229,  107, 1568,
          242, 2007,  719,  720,  610,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 397, 1050,  338, 1051, 1052, 1051, 1053, 1054,  324, 1055, 1056, 1057,
          397, 1051, 1053, 1054, 1058, 1059,  217,  397,  338, 1051, 1052, 1051,
         1053, 1054, 1060, 1061,  121, 1062,  102, 1063,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   33,   35,   35,  425,   16,   33,   18,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,  229,  116,   16,   29,   18,   16,   17,   34,  116,   16,   29,
           18,   16,   17,   34,  116,   16,   29,   18,   16,   17,   18,   18,
           18,  129,  593,  995, 1159,  997,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 824,  917,  918,  919,  920,  921,  920,  922,  619,  920,  923,  924,
          925,  920,  249,  926,  918,  924,  927,  409,  202,  920,  313,  927,
          918,  333,  202,  928,  929,  918,  920,  930,  918,  931,  923,  924,
          932,  218,  216,  919,  920,  922,  933,  934,  923,  924,  433,  231,
          935,  216,  919,  920,  934,  923,  924,  433,  942,  919,  920,  934,
          923,  924,  433, 1796,   71,  301,  973,  919,  920,  922, 1797,  934,
          923,  924,  433,  131,  918,  919,  920,  918,  923,  924,  433,  229,
          973, 1101, 1798,  142,  973,  859,  131, 1661,  900,  131,  111,  918,
          923,  924,  828,  938, 1799,  947,  131,  511,  131,   85, 1800,  969,
          225,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,  116,   16,  230,   17,   13,   18,   16,   29,   19,
          114,   13,   18,  797,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  379,    9,   28,  228,   20,   65,  228,   20,  145,  151,  152,
          234,  141,  358,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,  151,  226,    9,   28,   65, 1856,  125,  170,  162,    7,  145,
          151,   53,  116,   16,  125,   15,   16,   17,   18,   18,   16,   20,
          230,    9,   18,  229,  230,   28,  230,   65, 1857,  125, 1858,  151,
          713,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 686,  687,  688,  520,  688,  689,  558,  690,  691,  102,   68,  464,
          692,  102,   68,  465,  690,  693,  102,   68,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1768, 2070,  256,  579, 2071, 2072,  135,  579, 2073, 2074, 1274,
          260,  579, 2075,  330,  328,  710, 1768, 2075, 2076, 2077, 1072, 2078,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,   92,   16,   29,   18,  273,  178,
           11,   12,  160,  161,   16,  277,   18,   13,   10,  114,   15,   16,
          278,   18,  279,  164,   79,   10,  230,   15,   16,  278,   18,  279,
          166,  161,   24,  167,  279,  273,  140,  141,  280,   13,  281,  282,
          283,  284,  285,  272,  173,   65,   92,   16,   17,   18,  286,  178,
           21,  287,  114,  288,   19,  182,  287,  230,  288,   34,   20,   10,
           23,  278,   10,   29,  249,   65,   92,   16,   29,   18,  173,  278,
           10,   21,  249,  642,   65,   92,   16,   29,   18,   65,   92,   16,
           17,   18,  289,   68,  274,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,  113,   13,   91,   19,  887,   26,   34,  887,   93,  251,  799,
         1066,  113,  822,   19,   29,  578,  113,   91,   13,   19,   17,   93,
          578,   29,   13,  254,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   26,   27,  889,   16,  170,   18,
           36,  889,   16,  125,   18,   36,   10,   29,   26,   27,  889,   16,
          125,   18,  263,  889,   16,  170,   18,   10,   19,   29,  889,   16,
          125,   18,  263,   91,   17,  889,   16,  125,   18,   19,  889,   16,
          170,   18,   93,   10,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  353,   92,  131,  370,  164,   16,  164,  181,  131,  183,  125,
           92,   16,   29,   18,   34,  125,   92,   16,   20,   18,   34,  125,
           92,   16,   30,   18,   10,   20,   26,  353,   92,   16,   30,   18,
           10,   91,   27,   93,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   52,   63,  102,  256,  260, 1088, 1009,  690,  455,  560,   10,
         1465,   66, 1569, 1570,  172,  855, 1571,  135,  241,   28,   65,  260,
          135,  960,  961,  153,   10,   29,  256, 1088,  466,  229,  811, 1572,
           53,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,   83,   16,  151,   18,  102,  114,  116,   16,   65,   18,   16,
           17,   18,   10,  116,   16,  228,   16,   30,   18,   18,   16,   30,
           18,   26,   27,  154,   10,   29,   26,   27,  155,   10,   30,  153,
           10,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1648, 1649, 1650, 1017, 1365,  702, 1650,  102,  530, 1651, 1652, 1017,
         1653,  702,   71,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  272,  273,  178,  231,  285,  272, 1469,  102,    9,   91,   17,
           26,   23,   93,   28,   11,   91,  228,   16,   17,   18,   26,  116,
           16,  117,   18,   16,   21,   18,   24,   93,   65,   11,   91,  228,
           16,   17,   18,   26,  116,   16,   20,  117,   18,   16,   21,   18,
           24,   93,   66,   91,   17,   26,  117,   93, 1470, 1245,   16,  153,
           18, 1245,   16,  154,   18, 1245,   16,  269,   18,  104,  234,   11,
           91,   29,   26,  116,   16,  117,   18,   16,   17,   18,   24,   93,
           91,   29,   26,  117,   93,  173,  464,   92,   16,   29,   18, 1470,
         1245,   16,  153,   18,  173,  464,   92,   16,   17,   18, 1470, 1245,
           16,  154,   18,  173,  464,   92,   16,   20,   18, 1470, 1245,   16,
          269,   18, 1269,  464,   92,   16,   29,    0],
        [ 274,  275,  276,  102,  140,  141,  280,   13,  456,  580,  282,  283,
          284,  285,  272,  173,   65,   92,   16,   29,   18,  286,  178, 1618,
         1513, 1392,  464,  173,   65,   92,   16,   29,   18,  265,   68,  142,
         1089,  848,   36,  848,   36, 1619,   36,  144,   36,  182, 1180,  142,
          512,   65,   92,   16,   17,   18,  274,  178, 1270,    9,  286,  116,
           16,  117,   18,   16,   20,   18,   68,   28,  173,   65,   92,   16,
           17,   18,  229,  145,  956,   53,  168,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 242,    9,   28,  138,  270,   65,  265,  261,   16,  155,   18,  263,
          261,   16,  154,   18,   10,   29,   65,  512,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1358,   13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   17,
           13,   34,   20,   79,   19,  105,  165,   23,  164,   13,   34,   79,
           19,   20,  163,   23,  164,   79,   19,   17,  163,   23,  166,  161,
           24,  167,   33,  807, 1647,   79,  168, 1667,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  136,  106,  228,   20,  360,  178,   79, 1489,  116,   16,   29,
           18,   16,   17,   18,   13,   26,  136,  192,  178,  179,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 370,   11,   12,  125,   92,   16,  131,   18,   24,   25,  181,  131,
          183,   83,   16,  353,   18,   92,   16,  131,   18,   26,   83,   16,
          125,   18,   92,   16,   20,   18,   10,   20,   26,   83,   16,  353,
           18,   92,   16,   21,   18,   10,   96,  424,   92,   16,  278,   10,
           29,   18,   15,   16,  131,   18,  116,   16,   29,   18,   16,  353,
           92,   16,  278,   18,   18,   10,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 482, 2186, 2187,  614, 2188, 2189,  417, 2190, 2191,   96,   96, 2192,
         2193, 2194, 2195, 2196,  417, 2197,  635, 2198,  614,  636,  212,  209,
         1206,  612, 2199,  614,  944,  212, 2200, 2201, 1990,   71,  900,  209,
         2200, 2201, 1990,   71, 2202, 2203,  880, 1734, 2190,   96,   96, 2192,
         2204, 2200,  632,  973, 2205, 2198,  614,  636,  229,  142,   91,  973,
           10,   17,   93,  229, 1455,  973,  212,  635,   71,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1338,  102, 1171, 1339, 1926,   71,  373, 1339, 1926, 2023,  880, 1734,
           61,  973, 1338,   96, 1409, 1339,  102, 1926, 1275,   83,   16, 2024,
           18,   10, 2025,   26,   27,   83,   16,  142,   18,   91,   83,   16,
           13,   18,   10,   21,   93,   22,   83,   16,  142,   18,   91,   83,
           16,  973,   18,   10,  105,   93,   26,   27,   83,   16,  373,   18,
           10,   91,   27,   93,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  129,  229,  794,   20,   15,   16,  131,   18,   19,   17,   15,
           16,  131,   18,  257,  993,  122,  994,  131,  138,  995,  996,  997,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,   92,   29,  273,  178,   11,   12,
          160,  161,   16,  277,   18,   13,   10,  228,   16,   20,   18,  114,
          384,  164,   79,   10,  230,  384,  166,  161,   24,  167,  384,  273,
          140,  141,  280,   13,  456,  580,  282,  283,  284,  285,  272,  173,
           65,   92,   17,  286,  178,  287,  230,   11,   91,  288,   34,  116,
           16,  117,   18,   16,   21,   18,   24,   93,   10,   17,  228,   16,
           17,   18, 1269,   65,   92,   29,  178,   65,   92,   17,  274,  178,
         1270,  142,   65,   92,   29,   68,  568,   65,   92,   17,  229,   36,
         1256,   36,  111,  142,  274,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 61, 112, 113,  91,  13,  93,  10, 174,  13,  19,  13,  34,  29, 250,
         113,  91,  13,  93, 251, 252, 241,  13, 294,  91,  29,  26,  34, 887,
          93, 249,  26,  29,  22, 116,  16,  13,  19,  29,  18,  16, 174,  13,
          18,  22,  13,  61, 162, 257,  29, 241,  13, 294,  91,  23,  26,  29,
          93, 249,  29,  34,  91, 162,  19,  29,  93,  13, 257, 162,  15,  13,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7,   8,   9,  10,  12,  13,  36,  13,  22,  29,  25,  26,  27,  28,
          10,  11,  12,  13,  36,  20,  15,  16,  13,  18,  22,  29,  24,  25,
          26,  27,  27,  36,   4,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 61, 112, 113,  91,  13,  93,  10,  36,  17,  13,  34,  29,  36,  34,
          36,  13,  19,  29,  36, 292,  79,  10, 113,  91,  13,  93, 797,  13,
         294, 118,  23,  26,  34, 887,  93, 249, 113,  91,  13,  93, 578, 641,
          34, 170, 229, 125, 170, 111,   3,   3,   3,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [127, 231, 129, 355, 125,  10,  19,  29, 130, 353,  10,   3,   3,   3,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   10,   12,   29,   26,   20,   26,   30,   26,  133,
           25,   26,   28,   10,   12,   17,   26,   20,   26,   21,   26,   30,
           25,    9,   31,   28,   10,   91,   27,   93,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  272,  273,  178,  274,  275,  276,  102,  173,   65,   92,   29,
          273,  178,   11,   12,  160,  161,   16,  277,   18,   13,   10,  114,
           15,   16,  278,   18,  279,  164,   79,   10,  230,   15,   16,  278,
           18,  279,  166,  161,   24,  167,  279,  273,  140,  141,  280,   13,
          281,  282,  283,  284,  285,  272,  173,   65,   92,   17,  286,  178,
           21,  287,  114,  288,   19,  182,  287,  230,  288,   34,   20,   10,
           23,  278,   10,   29,  249,   65,   92,   29,  173,  278,   10,   21,
          249,  229,   65,   92,   29,   65,   92,   17,  289,   68,  274,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  248,   15,   13,   91,   17,
           13,   19,   29,   93,   19,  641,   34,  125,  125, 1737,   29, 1181,
         1546,   13,   92,   23,  113,   91,   13,   92,   23,   93, 1737,   23,
          125,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  290,  291,   61,  125,   26,  170,   26,  162,  294,  573,   26,
          125,   34,  170,   34,  162,   10,   23,   26,  574,   10,   29,  241,
          575,  540,  576,  756,   16,  125,  170,  162,  125,  170,  162,  168,
          241,  577,   12,  125,   26,  170,   26,  162,   25,  165,  228,  118,
           20,  119,   16,   21,   18,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1200, 1201, 1202,  249, 1203, 1204, 1205,  181, 1206, 1207, 1208,  464,
          502,  465,  102, 1209, 1210,  102,  733, 1200,  355, 1205, 1211, 1204,
           71,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1594, 2000, 2001,  180,   11,   12,  125,   92,   16,  131,   18,   24,
           25,   11,   12,  125,   92,   16,  131,   18,   24,   25,  682, 2002,
          183,  171, 2003,  171, 2003,  378,   83,   16,  278,   18,  578,   17,
           83,   16,  171,   18,   26,   83,   16,  125,   18,   92,   16,   29,
           18,   26,   83,   16,  125,   18,   92,   16,   17,   18,   26,  721,
           26,   83,   16,  125,   18,   92,   16,   83,   16,  278,   18,   18,
          102,  307,  307,  171, 2000, 2001,  180,  682,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  750,   63,   19,    9,   92,   16,   29,   18,   28,   92,   16,
           29,   18,   65,   92,   16,   29,   18,   66,   92,   16,   29,   18,
           59,   63,   52,   68,  256,  558,  468,   92,   16,   29,   18,  814,
          172,  821,   92,   16,   29,   18,  241,  814,  815,  242,  472,   92,
           16,   29,   18,   65,   92,   16,   29,   18,  603,    9,   92,   16,
           29,   18,  256,  229,  607,   28,   19,  821,   19,   65,   92,   16,
           29,   18,  720,  610,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  106,    9,  177,  278,  107,  277,  104,   65,   91,   13,   19,
           17,   93,   15,   16,   17,   18,   34,   91,   79,   19,   20,   93,
           15,   16,   17,   18,   10,   29,  515,   68,  464,  465,  362,  229,
          278,  254,  261,   16,   83,   16,  848,   18,   18,  263,  261,   16,
           83,   16,  849,   18,   18,   10,  298,   62,  140,  141,  229,   36,
          470,   36,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 173,   79,   10,   13,   15,   16,   17,   18,   34,  116,   16,   29,
           18,   16,   13,   18,   68,  990,   29,  584,   17,  991,  579,  176,
          178,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1114,  969, 1115, 1116,   96,  417,  445, 1117, 1118, 1119, 1120,  869,
         1121,  734, 1122, 1122,  670, 1121, 1123,  271,  272,  273,  178,  274,
          275,  276,  102,  173,   65,  273,  178,   11,   12,  160,  161,   16,
          277,   18,   13,   10,  116,   16,   29,   19,  279,   15,   16,   17,
           18,   18,   16,   29,   34,  279,   15,   16,   17,   18,   18,  164,
           79,   10,  116,   16,   21,  279,   18,   16,   29,   34,  279,   15,
           16,   17,   18,   18,  166,  161,   24,  167,  279,  273,  140,  141,
           62,  280,   13,  456,  580,  282,  283,  284,  285,  272,  107,  277,
          286,  178,   17,  373,  114,  288,   34,  228,   16,   20,   18,  287,
          230,  288,   34,  482,   10,   23,  229,   65,  277,  274,  178,  229,
           65,   68,  277,  358,  111,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  230,   36,   13,   36,   34,   36,
          230,   13,   36,  385,   99, 1062,  113,   13, 1222,  113,   13,  546,
           11,   91,  116,   16,  117,   18,   16,   17,   18,   26,  117,   24,
           93,  251,  888,  113,   13,  118,   19,  117,   26,  117,  119,  253,
          113,   13,  168, 1063, 1062, 1148,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   92,   16,   29,   18,   10,  116,   16,  117,   18,   16,   21,
           18,   13,   92,   16,   17,   18,   10,  116,   16,   20,  117,   18,
           16,   21,   18,  112,  113,   91,   13,   93,   10,  230,  115,   13,
           91,  115,  257,   23,   93,  138, 1397, 1113,   68,  115,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   91,   13,  294,  296,   16,  573,
           18,   93,  113,   91,   19,   13,   93,   10,   17,   19,  113,   91,
           13,   93,  112,   79,   10,  116,   16,   13,   34,   29,   18,   16,
           13,   18,   79,   10,  113,   91,   13,   93,  797,  267,   11,   91,
           13,   92,   16,   29,   18,   26,   79,   92,   16,   29,   18,   24,
           93,   26,   27,   11,   91,   13,   92,   16,   17,   18,   26,   79,
           92,   16,   17,   18,   24,   93,   26, 1026,   26,   11,   91,   13,
           92,   16,  171,   18,   26,   79,   92,   16,  171,   18,   24,   93,
          424,   92,   16,   35,   10,   29,   18,   15,   16,  171,   18,   11,
           91,   13,   92,   16,   35,   18,   34,   79,   92,   16,   35,   18,
           24,   93,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93, 1423,  573,  113,   91,   13,   34,
           29,   93,   10,   17,  113,   91,   13,   93,   13,  294,   91,   23,
           26,   29,  119,  249,  113,   91,   13,   93,   10,   13,   91,   13,
           19,   29,   93,  378,   13,  294,   91,   19,  887,   26,  171,  119,
          113,   91,   13,   93,  165,   19,  116,   16,  134,   18,   16,  366,
           18,  171,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  174,   36,   17,   13,   34,
           29,   36,   19,  174,   36,   17,   13,   19,   29,   36,  113,   91,
           13,   93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  114,   17,   13,   34,  105,  114,
           11,   91,  116,   16,  117,   18,   16,   17,   18,   19,   13,   24,
           93,  168,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 113,   91,   13,   93,   10,  114,   13,   19,  230,   13,  118,   19,
          125,   26,  125,  119, 1627,  112,  125,  168,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,   83,   16,  151,   18,  226,    9,   28,   65,  125,  170,  162,
           17,  170,  114,   28,   10,  125,  114,   65,   34, 1597,  114,    9,
           28,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  882, 1698, 1895,  558,  107,  242, 1393, 1074,  647, 1393, 1896,
          882,  877, 1065,   53,  168,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738,  121,  463, 1461,   26, 1462,  363,  151,  812, 1463,  242,  979,
          153,   10,   29,   26,  814,   10,  562,   10,   17,  239, 1464,   10,
          351,   15,   16,   82,   18,  153,   26,  154, 1084,  814,  562,  455,
          951, 1465,  231,  241,  121,  102,  960,    9,   26,   65,   26,  260,
           26,   66, 1116,  718,  242,  151,  172,  242, 1466, 1467,  102,  811,
         1468,   53,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301, 1424,  370,   16,  181,  131,  183,  125,   92,   16,   21,   18,
           34,  125,   92,   16,   30,   18,   10,  354,   26,   27,  353,   92,
           16,  105,   18,   10, 1697,   26,   11,   12,  125,   92,   16,  131,
           18,   24,   25,  795,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  180,   11,   12,  125,   92,   16,  131,   18,   24,   25,  181,
          131,  183,   83,   16,  353,   18,   92,   16,  131,   18,   10,   29,
           34, 1348,   83,   16,  125,   18,   92,   16,  131,   18, 1348,  572,
           23,  241,   11,   12,  125,   92,   16,  131,   18,   24,   25,  132,
          914,  526,  527,   83,   16,  353,   18,   92,   16,   30,   18,   10,
          116,   16, 2004,   18,   16, 2005,   18,  229, 1349,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  139,  135,   92,   16,   29,   18,   91,   19,   29,
           26,   23,   93,   26,  135,   92,   16,   17,   18,   91,   29,   26,
           23,   93,  135,   92,   16,   17,   18,  107,   65,  515,    9,   28,
          362,   11,   36, 1162,   92,   16,   17,   18,   24,   36,   10,   17,
           11,   36,  135,   92,   16,   17,   18,   28,   24,   36,   26,   36,
          153,   36,   10,   11,   36,  562,   92,   16,   29,   18,   24,   36,
           65,  178,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  174,   13,   34,  641,   15,
           16,   17,   18,   34,   91,   17,  125,   34,   29,   93,   13,  250,
          113,   91,   13,   93,  251, 1316,  125,   22,   23,  249,   26,  241,
          113,   91,   13,   93,  163,   19,  116,   16,   20,   18,   16,   21,
          125,   18,   19,   17,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,   68,  465,   52,   63,   64, 1412, 1413,  236,  242, 1413,  815,
          242,   63,  464, 1089,  561,  466,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  174,   13,   19,  116,   16,
           13,   34,   29,   18,   16,   13,   19,   29,   18,  250,  113,   91,
           13,   93,  251,  252,  241,  113,   91,   13,   93,  380,  381,  138,
          253,   61,   13,   92,   16,   23,   18,  113,   91,   13,   93,  253,
          241,  173,   79,   10,  174,   13,   68,    9,   11,   91,   13,   92,
           16,   23,   18,   26,  174,   13,   92,   16,   23,   18,   24,   93,
          579,  176,  173,   79,   10, 2206,   16,  248,   18,   15,   16,   13,
           18,  176,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 368,  369,   17,   15,   16,   13,   18,   19,   17,   15,   16,   79,
           18,   22,   20,   15,   16,   19,   13,   18,   19,   20,   15,   16,
           19,   79,   18,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,   83,   16,  151,   18,  226,    9,  584,   28,  584,   65,  125,
          584,  170,  584,  162,    7,   83,   16,  170,   18,  230,   83,   16,
           65,   18,   34,   83,   16,  162,   18,  230,   83,   16,   28,   18,
           10,   21,   83,   16,  125,  886,   28,   18,  230,   83,   16,   65,
           18,   26,   27,   83,   16,  170,   18,   15,   16,   17,   18,   34,
           83,   16,  162,   18,   15,   16,   17,   18,   19,   83,   16,  125,
           18,   15,   16,   17,   18,   10,  134,  145,   83,   16,  151,   18,
           53,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,   68,    9,  584,   28,  140,  141,   62,  736,   36,  153,   36,
           10,    9,  150,  464,  106,    9,  584,   28,  107,   13,   34,   17,
           10,   23,  357,    9,  107,   13,   34,   79,   10,   23,  229,  150,
          464,  523,  270,  142,    9, 1182,  249,   36, 1183,   36,   19,   36,
         1184,   36, 1185,  449,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10, 1112,   15,   16,   13,   18,
           19,  174,   13,   19,   29,   61,   13,   10,   17,  113,   91,   13,
           93, 1113,   68,  229,  125,  914,  113,   91,   13,   93,  251,  546,
          241,  125,  589,  116,   16,   29,   18,   16,  248,   18,  249,  113,
           91,   13,   93,  589,   23,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  135,  136,  162,  137,   13,   15,   16,   17,   18,   19,  116,
           16,   79,   15,   16,   17,   18,   18,   16,  134,   18,   10,   29,
          259,  139,  142,   65,  258,  808,  238,    9,   91,   23,   26,  105,
          228,   16,  105,   18,   93,  145,   83,   16,  809,   18,  713,  588,
           56,   53,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   11,   12,  160,  161,   16,
         1079,   18,   13,   34,   29,   26,  744,   13,  163,   23,  164,   17,
           15,   16,   13,   18,   26,  744,   13,  257,   23,  166,  161,   24,
          167,  113,   91,   13,   93,   34,  113,   11,   91,   13,   19,  116,
           16,   29,   18,   16,   17,   18,   24,   93,  257,   29,   13,  254,
         1533,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7, 1223, 1165, 1186,  132,   11,   12,  125,   92,   16,  131,   18,
           24,   25,  181,  183,  483,  125,   92,   16,   30,   18,   10,   20,
          125,   92,   16,  170,   18,   34,   21,  125,   92,   16,   29,   18,
          125,   92,   16,   20,   18,   10,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   11,   12,  160,  161,   16,
         1079,   18,   29,   34,  126,   92,   16,   17,   18,   91,   17,   19,
           13,   93,   26,  744,   13,   22,   29,  164,   17,   15,   16,   13,
           19,   29,   18,   26,  744,   13,  589,   29,  166,  161,   24,  167,
          113,   91,   19,   17,   93,   34,  113,   11,   91,  126,   92,   16,
           17,   18,  298,   24,   93,   10,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  232,  233,  353,  524, 1003, 1004,  719,  609,  741,  116,   16,
          133,   18,   16,  134,   18, 1003,  232,   59, 1074, 1075,  703,   15,
           16,   82,   18,  145,   83,   16, 1629,   18,   53,   30,  228,   16,
          483,   18,  232,  245,   53,  179,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   11,   12,  125,   92,   16,  131,   18,   24,   25,  795,  370,
           83,   16,  353,   18,   92,   16,  131,   18,   11,   12,  125,   92,
           16,  131,   18,   24,   25,  181,  131,  183,   83,   16,  353,   18,
           92,   16,  134,   18,   10,   21,   83,   16,  353,   18,   92,   16,
           21,   18,  125,   92,   16,   96,   18,   10,   91,   27,   93,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,  102, 1423, 1531,  112,   79,   10,   96,   15,   16, 1532,   13,
           18, 1423, 1531,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 363,   83,   16,  151,   18,  226,    9,   26,   28,   26,   65,  125,
           26,  170,   26,  162,   26,   61,   91,  230,   28,   19,  230,   65,
           93,   15,   16,   17,   18,   10,  230,   15,   16,   17,   18,    9,
           19,  230,   28,  230,   65,  229,    9,  228,   16,   17,   18,  125,
           34,  170,   10,   17,  162,  584,  229,  230,   65,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   68,    9,   28,  169,  261,   16,   83,   16,  155,   18,   18,
           10,   91,   19,   21,   26,   19,   20,   93,  169,  261,   16,   83,
           16,  154,   18,   18,   10,   91,   27,   93,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,   92,   16,   29,   18,  137,   11,
           12,  160,  161,   16,  277,   18,   13,   10,  279,  114,  384,  164,
           79,   10,  279,  230,  384,  166,  161,   24,  167,  279,  273,  279,
          572,   23,   23,  163,  384,  163,  117,   62,  280,   13,  281,  282,
          283,  285,  272,  102,  173,   65,   92,   16,   17,   18,  137,  287,
           10,   17,  230,  288,   65,   92,   16,   20,   18,  137,  287,   10,
           17,  228,   16,   20,   18,  114,  288,  229,   65,   92,   16,   17,
           18,   65,   92,   16,   20,   18,  267,  274,   65,   92,   16,   29,
           18,   65,   92,   16,   17,   18,  374,  853,    9,   65,   92,   16,
           29,   18,   65,   92,   16,   20,   18,  374,  853,   28,  229,   36,
          153,   36,  168,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,   91,   13,   93,   10,  641,   15,   16,   17,   18,   19,
          641,   19,   13,  174,   13,  113,   91,   13,   93,  589,   23,  229,
          125,  241,  113,   91,   13,   93, 1181, 1964,   68,   13,   92,   16,
           23,   18,   16,  248,   18,   15,   16,   19,   17,   18,   22,   16,
          113,   18,   11,   91,   16,   13,   18,   92,   16,   23,   18,   24,
           93,   22,   17,   15,   16,   19,   17,   18,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 763,  764,   40,  765,  766,  767,  768,  769,  770,  771,  226,  772,
          773,   58,  774,  775,  776,  777,  778,  779,  780,  781,  777,  231,
          777,  782,  232,  783,  777,  782,  784,  785,  786,  777,  782,   50,
          786,  787,  782,  247,  781,  777,  788,    7,  789,  777,  247,  510,
          790,  791,  786,  792,  510,  793,  794,  781,  777,  510,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,   17,  230,   11,   91,   17,   13,   34,  116,   16,
          117,   18,   16,  105,   18,   24,   93,  797,  259,  876,  116,   16,
           29,   18,   16,   21,   18,  123,  877,  797,  112,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  107,  277,   92,   16,   29,   18,  273,  178,
           11,   12,  160,  161,   16,  277,   18,   13,   10,   17,   34,  279,
          164,   79,   10, 1317,  166,  161,   26,   27,   91,  279,  478,   16,
          479,   18,   93,   24,  167,  107,  277,   92,   16,   17,   18,  273,
          178,   11,   12,  160,  161,   16,  277,   18,   83,   16,   13,   18,
           10,   19,   17,   34,   83,   16,  171,   18,  164,   83,   16,   79,
           18,   10,  116,   16,   83,   16,  171,   18,   18,   16,   83,   16,
          278,   18,   18,   27,   26,   27,   91,   83,   16,  171,   18,  478,
           16,  479,   18,   93,  167,  478,   16,   61,   18,  277,   92,   16,
           29,   18,  478,   16, 2020,   18,  277,   92,   16,   17,   18,  478,
           16, 2021,   18,   83,   16,  142,   18,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 274,  275,  276,  102,  173,   65,   92,   29,  137,   11,   12,  160,
          161,   16,  277,   18,   13,   10,  885,   16,  279, 1513,   18,  384,
          164,   79,   10,  521,  384,  166,  161,   24,  167,  279,  273,  279,
          572,   23,  991,   23,  578,  384,  578,  117,   62,  280,   13,  281,
          282,  283,  285,  272,  102,  173,   83,   16,   65,   18,   92,   16,
           17,   18,  137,  287,   10,   17,  230,  288,   83,   16,   65,   18,
           92,   16,   20,   18,  137,   27,  287,   10,   17,  228,   16,   20,
           18,  114,  288,  229,   65,   92,   29,   65,   92,   17,  267,  274,
           65,   92,   29,   65,   92,   17,  374,  853,    9,   65,   92,   16,
           29,   18,   65,   92,   17,  374,  853,   28,  229,   36,  153,   36,
          168,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,  116,   16,   17,   13,   15,   16,   20,   18,   18,
           16,   17,   15,   16,   13,   18,   34,   17,   15,   16,   19,   13,
           18,   18,  120,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  135,  136,   65,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,
           16,   17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,
           29,  125,  170,  259,  139,   62,  140,  141, 1416,  604,  104,  104,
           13,   15,   16,   17,   18,   34,   79,   15,   16,   17,   18,   10,
          125,   15,   16,   17,   18,  515,  142,  568,  362,   36, 1256,   36,
           36, 1416,   36,   65,  457,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93, 1594,  573, 1066,   13,  294,   91,
           19,  887,   26,   23,   93,  249,  113,   91,   13,   93,   10,   17,
           13,   15,   16,   20,   18,   34,   13,   15,   16,   17,   18,  113,
           91,   17,   93,  179,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 396,  397,  398,  399,  400,  401,  402,  403,  404,  405,  397,  406,
          402,  403,  407,   53,  408,  409,  410,  410,  102,  411,  412,  413,
          414,   85,  415,  405,  416,   90,   11,   91,   13,   92,   16,   35,
           18,   26,   79,   92,   16,   35,   18,   24,   93,   91,   35,   10,
           29,   26,   17,   26,   94,   26,   85,   93,   26,   13,   92,   16,
           35,   18,   79,   92,   16,   35,   18,  417,   35,  418,  419,  420,
          421,   81,  422,  402,  403,  423,  424,   92,   16,   35,   10,   29,
           18,   15,   16,   85,   18,   13,   92,   16,   35,   18,   10,  351,
           27,   26,   27,  424,   92,   16,   35,   10,   29,   18,   15,   16,
           85,   18,   79,   92,   16,   35,   18,   10,  318,   27,   26,   27,
          424,   92,   16,   35,   10,   29,   18,    0],
        [  61,   62,  140,  141,  107,   13,   10,  125,  136,   65,  116,   16,
           13,   15,   16,   17,   18,   18,   16,  125,   15,   16,   17,   18,
           18,   19,  116,   16,   79,   15,   16,   17,   18,   18,   16,  170,
           15,   16,   17,   18,   18,   10,   29,   91,  125,  257,   23,   26,
          170,  257,   23,   93,  359,  360,  361,   66,  256,  362,  145,  364,
           53,   65,  365,  111,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 314,  917, 1194, 1286,  447, 1287, 1288, 1289, 1290, 1291,  202, 1292,
         1291,  202, 1094, 1293, 1287, 1294, 1291,  202, 1295,  534, 1296, 1297,
         1298, 1299, 1300,  534, 1192,   81, 1192, 1301,  214, 1300,  333, 1302,
         1192, 1274,  546,   85,  214, 1300, 1303, 1302, 1192, 1301,   85, 1300,
          409, 1302, 1304, 1286,  917,  217,  181,  922, 1304, 1305, 1192,   90,
         1100, 1101, 1102, 1192, 1274,  546,  942, 1192, 1274,  546,   71,  229,
         1304, 1287, 1300,  973,   81, 1302, 1101, 1798,   61, 1304, 1298, 1287,
          832, 1306,   81,  202, 1304, 1287, 1288,  131,   81, 1302,  249, 1306,
          766, 1866,  168,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 417,  463,   63,  102,  153,  594,  265,  142, 2010,   65,  234,  813,
          357, 2011,  261,   16,   83,   16, 1963,   18,   18,   10, 1348,  261,
           16,   83,   16,  153,   18,   18,   34, 1868,  261,   16,   83,   16,
          594,   18,   18, 1348,   34, 1868,  168,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  91,   13,   34,   79,   93,   91,   17,   13,   19,   79,   93,   15,
           16,   30,   18,  582,  102,   13,   15,   16,   20,   18,   79,   15,
           16,   20,   18,  583,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1534, 1535, 1536,  212,  209, 1537, 1538, 1539, 1540, 1535,  212,  209,
         1535, 1541,  209, 1535,  733, 1542, 1543, 1535, 1541, 1543, 1535,  733,
         1544, 1535,  733,  212, 1535,  733,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  116,   16,  248,   15,   16,
           13,   18,   18,   16,   13,   34,  125,   18,   26,  113,   15,   16,
          522,   18,   91,   29,   93,   10,  116,   16,  248,   18,   16,   21,
           18,   26,  125,   10,    4,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,   17,   13,   15,   16,   17,   18,   19,  248,   15,
           16,   36,   13,   36,   18,  822,  797,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    9,  513,   65,  137,   79,   15,   16,   17,   18,   10,   17,
          514,   91,  373,  257,   23,   93,  238,   68,    9,   65,  139,  358,
          298,   79,  456,  358,  373,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   10,   12,   29,   26,   17,   26,   20,   25,   26,
           28,   10,   12,   17,   26,   20,   26,   21,   25,    9,  644,   28,
           10,   91,   27,   93,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   36,   13,   34,   29,   36,
           19,   36,   17,   13,   19,   20,   36,  839,  121,  964,  794,   79,
           10,  113,   91,   13,   93,  797,  842,  229,  290,   36,  113,   91,
           13,   93,   36,  257,   29,  293,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1213, 1214,   52,   51, 1215, 1216,  292,  688,  520,  688,  158,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,   83,   16,  135,   18,   92,   16,   29,   18,   26,   83,   16,
          135,   18,   92,   16,   17,   18,  255,   65,  138,  139,  142,   65,
          238,   83,   16,  146,   18,   92,   16,   29,   18,  172,   83,   16,
          146,   18,   92,   16,   17,   18,  239,   83,   16,  146,   18,   92,
           16,   17,   18,   83,   16,  135,   18,   92,   16,   29,   18,   10,
          351,   15,   16,   82,   18,   65,  457,   91,   27,   93,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738,   61,  180,   11,   12,  125,   92,   16,  131,   18,   24,   25,
          125,   92,   16,   29,   18,   10,   20,   26,  125,   92,   16,  131,
           34,   29,   18,   10,   20,  125,   92,   16,  131,   18,   19,   21,
          131,  423,  125,   92,   16,   29,   18,   26,  125,   92,   16,   17,
           18,  739,   11,   12,  125,   92,   16,  131,   18,   24,   25,  526,
          527,  241,  229,  180,   11,   12,   17,   15,   16,  131,   18,  125,
           92,   16,  131,   18,   24,   25,  181,  131,  183,  353,   92,   16,
          131,   18,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301,   83,   16,  353,   18,   92,   16,  131,   18,  132,   11,   12,
          125,   92,   16,  131,   18,   24,   25,  181,  131,  183,    7,   83,
           16,  353,   18,   92,   16,   17,   18,   10,   17,   26,   27,   83,
           16,  353,   18,   92,   16,   20,   18,   10,   19,  105,  229,   11,
           12,  125,   92,   16,  131,   18,   24,   25,  526,  527,  229,   83,
           16,  353,   18,   92,   16,  131,   18,  554,  353,   92,   16,  131,
           34,   29,   18,   26,   27,  353,   92,   16,  131,   18,   26,   27,
          353,   92,   16,  131,   34,   17,   18,  985,  370,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93, 1222,   13,   22,   23,  249,  113,   91,
           13,   93,   10,  174,   91,   19,   13,   93,   34,   20,   13,  173,
           79,   10,  113,   91,   13,   93,   68,  579,  176,  178,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,   83,   16,  151,   18,  226,   83,   16,    9,   18,   26,   27,
           83,   16,   28,   18,   26,   27,   65,   83,   16,  125,   18,   26,
           27,   83,   16,  170,   18,   26,   27,   83,   16,  162,   18,  167,
           27,  145,   83,   16,  151,   18,   53,  116,   16,   83,   16,  125,
           18,   15,   16,   17,   18,   34,   83,   16,  170,   18,   15,   16,
           17,   18,   19,   83,   16,  162,   18,   15,   16,   17,   18,   18,
           16,   21,   18,  164,  164, 1777,   16,   65,  164, 2026,  164,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  124,  125,   10,  126,   92,   16,   20,   18,   17,   26,  170,
           10,  126,   92,   16,   30,   18,   20,   26,  162,   10,  116,   16,
           17,   18,   16,   20,   18,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  232,  233,  353,  524, 1003, 1004,  352, 1003,  232,   59, 1074,
         1075,  596,   15,   16,   82,   18,  145,   83,   16, 1629,   18,   53,
          232,  247,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1536, 1587, 1535,  102,  530,  414, 1587, 1588,  530,  414, 1587, 1589,
         1590, 1535,  433,  754, 1591, 1535,  433,   71,   91,   27,   93,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,   32,   33,   91,   33,   19,   29,   93,   27,   35,   10,   29,
           34,   35,   33,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  68,  107,   79,   10,  278,   91,   13,   34,   29,   93,  358,  168,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  226,    9,   28,   65,  125,  170,  162,    7, 1680, 1681,
           28, 1667,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,    7,  459,  460,  151,   19,    9,   92,   16,   29,   18,   28,
           92,   16,   29,   18,   65,   92,   16,   29,   18,   59,  236,  461,
          462,   92,   16,   29,   18,   65,   92,   16,   29,   18,   65,  463,
          464,   26,  465,  154,   26,   28,   92,   16,   29,   18,   65,   92,
           16,   29,   18,  466,  142,  467,  238,   28,   92,   16,   29,   18,
           65,   92,   16,   29,   18,  142,  242,  361,  153,  256,   26,  361,
          155,  135,  241,  468,   92,   16,   29,   18,  469,  469,  470,   26,
          242,    9,   92,   16,   29,   18,  471,  172,  242,  472,   92,   16,
           29,   18,   65,   92,   16,   29,   18,  135,   61,   62,  363,    9,
           92,   16,   29,   18,   28,   92,   16,   29,   18,   65,   92,   16,
           29,   18,   64,  473,   10,  153,   10,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1534, 1535, 1536,  212,  209, 1537, 1538, 1539, 1540, 1535,  212,  209,
         1535, 1541,  209, 1535,  733, 1542, 1543, 1535, 1541, 1543, 1535,  733,
         1544, 1535,  733,  212, 1535,  733,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  647,  384,  233,  140,  141, 1076,   13,  456, 1077,  282,  455,
         1078, 1067,  362,    9,   91,   29,   26,  125,   93,   26,   28,   27,
           91,   17,   26,  170,   93,  114,   17,  125,   10,  116,   16,   17,
           18,   16,   20,   18,   36,  125,   19,  170,   36,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  113,   13, 1066,   13, 1661,  249,  113,   91,   13,   93,   10,
           83,   16,  248,   18,   15,   16,   13,   18,   19,   29,   13,  249,
          113,   13,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  91,   29,   34,   35,   93,   91,   17,   34,   35,   93,   10,   91,
           27,   93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  91,   29,   34,   35,   93,   91,   17,   19,   35,   93,   10,   91,
           27,   93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  116,   16,   29,   18,   16,
           20,   18,   13,   15,   16,   20,   18,   19,  125,   11,   91,   13,
           15,   16,   17,   18,   34,   13,   34,   29,   24,   93,  125,  229,
          113,   91,   13,   93,  251,  546,  241,  113,   91,   13,   93,  253,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   33,   10,  116,   16,   29,   19,   35,   18,   16,   29,   34,
           35,   18,   34,   17,   35,   36,   33,   36,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [2119,  190, 2120, 2121, 2122, 2123,  190,  102, 1731,  530,  987,  221,
         2124, 2125,  414,  216, 1729,  973, 1156, 2120, 2126, 2024,   10,  179,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,    9,  584,   28,  173,   65,  137,   27,   79,   10,  116,   16,
           13,   15,   16,   17,   18,   18,   16,   21,   18,  362,    9,   28,
         1046,   21,  229,  107,  153,  177,   61,  464,  173,   65,  238,   65,
          464,  579,  176,  107,  153,  382,   83,   16,  467,   18,  172,   83,
           16, 1047,   18,  229,  107,  153,  178,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  800,   50,  138,   59, 1283,  604, 1284,  157,  148,  800,  247,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   62,  957,  669,  151,  238, 1511,   62,  145,  151,   57,  154,
          515,  464,  465,  362,   57,   58,  594,  515,   68,  260,  153,  155,
          357,  256,  135,  362,  241,  564,  154,  262,   54,  150,   62,  523,
          603,  470,   17,  228,   20,  229,  811, 1512,   53,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301,  353,   92,   16,  131,   18,  132,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,  125,   92,   16,   29,   18,
           10,   29,   26,   27,  353,   92,   16,   20,   18,   10,  116,   16,
           20,   18,   16,   21,   18,  353,   92,   16,   21,   18,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1166,  441, 1167,  530,  405,  203,  705, 1168, 1169,  203, 1170, 1169,
         1171, 1169, 1166, 1172,  446, 1173, 1173, 1174,  553,  450,  503,   15,
           16,   17,   18,   10,  116,   16,  131,   91,  539,   19,  540,   93,
           15,   16,   17,   18,   18,   16,   91,  125,   34,  170,   93,   91,
          162,   34,  541,   93,   91,  125,   34,  162,   93,   91,  170,   34,
          541,   93,   18,  219, 1168, 1175, 1169, 1166, 1176,   71,  555,  556,
         1168, 1175, 1169, 1166, 1172, 1174,  440,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  230,  384,   34,  114, 1750,   10,   29,   26,   27,  114,  384,
           34,  230, 1750,   10,   23,  230,   91,  384,   34, 1750,   93,   10,
            4,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   66,  232,  233,   62,  232,   59,  234,  145,  151,   59,  235,
          236,  142,  237,  238,  239,  240,   10,  201,   15,   16,   82,   18,
          241,  242,  243,  172,  242,  244,   61,  237,   10,  228,   16,   17,
           18,  232,  245,   53,  228,   16,   20,   18,  117,  229,  246,  142,
           19,  151,  247,    3,    3,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,  273,  178,   11,   12,  160,  161,
           16,  277,   18,   13,   10,   20,  114,  288,  164,   79,   10,  230,
          288,  166,  161,   24,  167,  288,  273,  107,  277,  273,  178,   11,
           12,  160,  161,   16,  277,   18,   13,   10,  125,   34,   21,  279,
          164,   79,   10,   29,   19,  279,  166,  161,   24,  167,  279,  273,
          125,   10,   19,   29,  229,   65,  277,  267,  140,   65,   68,  277,
          358,  168,  228,   16, 1048,   18,  229,  125,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  177,  278,  107,  277,  255,   65,  137,  116,   16,   13,   15,
           16,   17,   18,   18,   16,   21,   18,   34,  116,   16,   79,   15,
           16,   17,   18,   18,   16,   20,   18,   10,   29,  361,    9,   26,
           27,   28,  362, 1089,  153,  466,  464,   91,   29,   26,  171,   93,
           91,  171,  257,   23,   93,  241,  278,   22,   19,  116,   16,   29,
           18,   16,   17,   18,   61,  135,   65,  259,  139,  142,   65,  238,
          261,   16,   83,   16, 1390,   18,   18,   34,  261,   16,   83,   16,
         1391,   18,   18,   34,  261,   16,   83,   16,  563,   18,   18,   10,
          261,   16,   23,   18,  167,   27,  241,  137,   27,   36,  261,   16,
           83,   16, 1391,   18,   18,   36,   26,   36,  261,   16,   83,   16,
         1390,   18,   18,   36,   26,   36,  261,    0],
        [ 231,  812,   63, 1559,  155,  813,  515,   68,   62,   68,  256,  135,
          594,  269,  603, 1560,  564,  361,  813,  853,  952,  363, 1561,  564,
         1562,  363,   66,  851,  564, 1087,  241,  155,  172, 1563,   15,   16,
          522,   18,  153,   10,   30,   26,   27,  155,   10,  105,   26,   27,
          603,   10,  116,   16,   30,   18,   16,   21,   18,   26,   27,  516,
           15,   16,  522,   18,   10,   17,  228,   16,   17,   18,   26,   27,
         1564,   66,   15,   16,  522,   18,   19, 1565,  247,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  116,   16,  248,   15,   16,   13,
           18,   19,  248,   15,   16,   19,   13,   18,   18,   16,   13,   15,
           16,   17,   18,   18,  797,   91,   27,   93,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   11,   12,  125,   92,   16,  131,   18,   24,   25, 1223, 1165,
         1186,  132,  125,   92,   16,   29,   18,   10,   17,   26,  125,   92,
           16,   20,   18,   10,   17,  125,   92,   16,   17,   18,   34,  182,
          229,   11,   12,  125,   92,   16,  131,   18,   24,   25,  526,  527,
           61,  170,   92,   16,  131,   18,   10,  126,   92,   16,   17,   18,
          125,   92,   16,  131,   18,  229,  180,   11,   12,  170,   92,   16,
          131,   18,   24,   25,  181,  131,  183,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  353,   92,   16,  131,   18,  132,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,  125,   92,   16,   30,   18,
           19,  125,   92,   16,   20,   18,   10,  298,   26,   27,  125,   92,
           16,  105,   18,   19,  125,   92,   16,   21,   18,   10,  354,   26,
          116,   16,  353,   92,   16,  131,   18,   18,   16,  125,   92,   16,
          131,   18,   18,   10,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 125,  170,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   36,   13,   34,   29,   36,
           19,   36,   13,   19,   17,   36,  229,  290,  113,   91,   13,   93,
          589,   29,  293,  290,  113,   91,   13,   93,  589,   13,   15,   17,
           19,   13,   34,  171,  293, 1319,  229,  171,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,   92,   16,   29,   18,  137,  116,   16,   13,   15,
           16,   17,   18,   18,   16,  125,   15,   16,   17,   18,   18,   34,
          116,   16,   79,   15,   16,   17,   18,   18,   16,  170,   15,   16,
           17,   18,   18,   10,   29,   91,  125,  257,  170,  257,   23,   93,
          259,  139,  135,  454,   65,   92,   16,   17,   18,  139,  455,   26,
           27,   65,   92,   16,   29,   18,   64,   65,  233,  455,  135,   13,
          456,  352,  107,  361,   65,   92,   16,   29,   18,    9,   26,   28,
          362,  361,   65,   92,   16,   17,   18,   65,   66,  362,   36,  269,
           36,   10,  116,   16,   21,   18,   16,   20,   18,   36,  153,   36,
          229,   65,   92,   16,   29,   18,  457,   65,   92,   16,   29,   18,
           99,  233,   65,   92,   16,   17,   18,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 824, 1495,  190, 1496,    9,   28,  220,  397,  530,  405,   85, 1495,
         1495,  190, 1496, 1497,    9,  397, 2044, 2045, 2046, 2047, 2048, 2049,
         1715, 2050,  505, 2051, 2052, 2053, 2048, 2054, 2055, 2052, 2056, 2057,
         2051, 2058,   28,  397, 2045, 2059, 2044, 2060, 1282, 1154,  505, 2045,
         2050, 2057, 2061, 1697, 2062, 2046, 1715, 2063, 2064, 2051, 2062, 2065,
         1369,   90,  220,  397, 1495, 1496, 1497, 1376, 1377, 1376, 1377,  220,
          397, 1496, 1497, 1459, 1499,  975,  423,  794,  610,  446, 1062,  838,
         1495, 1496, 1497, 1495, 1496, 1500,   58,  193,  149,  194,  301, 1455,
           65,    9,  397, 1495, 1496,  194, 1557,   28,  397, 1495, 1496,  194,
          944,  220,  397, 1495, 1174,  880, 1734,   90, 1455, 1796,  942, 1455,
         1796,   71,  229,   65,   71,    0,    0,    0],
        [  61,   35,   13,  836,   13,   79,  861,   36,   13,   34,   79,   92,
           35,   36,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   10,   91,   19,   29,   26,   17,
           93,   26,  889,   16,  170,   18,   10,   91,  171,   26,   29,   93,
          169,  889,   16,  125,   18,   34,  889,   16,  170,   18,  352,  171,
           10,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   19,   13,   15,   16,   17,
           18,   34,  641,   34,   21,   26,   27,  477,   91,   13,   93,   10,
           36,   13,   34,   29,   36,   34,   36,   13,   19,   29,   36,  125,
           10,   29,  249,  229,  290,  113,   91,   13,   93,  589,  477,   91,
           13,   93,  293,  290,  113,   91,   13,   93,  589,  477,   91,   13,
           93,  293, 1049,  118,  822,   29,  584,   29,  119,  229,  125,  254,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  125,
           92,   16,   29,   18,   10,  116,   16,   29,   18,   16,   21,   18,
           26,   27,  125,   92,   16,   20,   18,  125,   92,   16,   30,   18,
           10,   21,   27,   11,   91,  125,   92,   16,   21,   18,   19,   29,
           24,   93,   26,   27,  125,   92,   16,   17,   18,   10,   91,   93,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  230,   15,   16,   17,   18,
           13,  230,   17,   13,  250,  113,   91,   13,   93,   91,   23,   26,
          117,   93,  251,  252,  241,   36,  113,   91,   13,   93,   36,  163,
          116,   16,   20,  228,   16,   20,   18,   18,   16,  134,   18,  241,
          230,   15,   16,   17,   18,   13,  230,   15,   16,   17,   18,   17,
           13,  230,   15,   16,   17,   18,   21,   13,  721,  230,   15,   16,
           17,   18,   17,   15,   16,  131,   18,   13,  163,  116,   16,   20,
           15,   16,  131,   18,   18,   16,   21,   15,   16,  131,   18,   18,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  125,   15,   16,   17,   18,   18,   34,  116,   16,   79,   15,
           16,   17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,
           29,   27,   91,  125,  257,  170,  257,   23,   93,   26,   27, 1877,
           16,  960, 1878,   18,  142,   92,   16,   29,   18,   91,   29,   26,
           29,   93,   26,  142,   92,   16,   17,   18,   91,   23,   26,   29,
           93,  142,   92,   16,   20,   18,   11,   91,   19,   29,   26,  116,
           16,  228,   16,   20,   18,   18,   16,   17,   18,   24,   93,   26,
           27,  142,   92,   16,   21,   18,   11,   91,   29,   26,  116,   16,
          228,   16,   20,   18,   18,   16,   17,   18,   24,   93, 1879,  379,
          255,   65, 1508,   65,  178,   61,  107,    0],
        [   7,  112,  113,   91,   13,   93,   10,   36,   13,   34,   29,   36,
           19,   36,   17,   13,   19,   20,   36,  121,  964,  794,   79,  113,
           13,  797,  229,  290,   36,  113,   13,   36,  293,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  372,  373,  726, 1420,  131,  294,  465,  584,  131,   15,   17,
         1724,   17,   15,  131,  392,  373,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  641,   15,   16,   20,   18,
           34,   13,   34,   29,  797,   68,  113,  579,  176,  106,  125,  179,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,  273,  178,   11,   12,  160,  161,
           16,  277,   18,   13,   10,   17,   19,  279,   19,  279,   15,   16,
           17,   18,  164,   79,   10,   17,   19,   20,  279,   34,  279,   15,
           16,   17,   18,  166,  161,   24,  167,  279,  273,  279,  572,   29,
           65,  356,  361,    9,   26,   28,  362,  229,   36,  153,   36,  140,
          141,  280,   13,  281,  282,  283,  284,  285,  272,  229,  107,  153,
          286,  178,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 882,  242,  801,  802,  631,  520,  231,  801,  247,  631,  247,   60,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,    8,  353,   10,   12,   13,   36,   91,   13,   19,   17,   93,
           91,   13,   19,   20,   93,  589,   23,   25,  499,   10,   12,   13,
           36,   13,  257,   23,   25,   83,   16,  353,   18,   31,   83,   16,
          499,   18,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 114,   11,   91,  116,   16,  117,   18,   16,   21,   18,   19,  384,
           24,   93,   10,  116,   16,   20,   18,   16,   30,   18,  230,   17,
          384,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   26,   79,  159,   11,   12,  160,  161,   16,  162,   18,   17,
           13,   34,   79,   19,   17,  163,   23,  164,   13,   19,   79,   19,
           29,  165,   23,  164,   79,   34,   29,  165,   23,  166,  161,   24,
          167,   33,   10,   13,   34,  133,   79,  168,    4,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  169,  889,   16,  125,   18,   10,   91,  171,   26,   29,   93,
           26,  889,   16,  170,   18,   10,   91,   29,   26,   17,   93,   26,
           27, 1097,   36,  889,   16,  125,   18,   34,  889,   16,  170,   18,
           36,   15,   16,   17,   18,   10,   36,  889,   16,  125,   18,   36,
           15,   16,   17,   18,   34,   36,  889,   16,  170,   18,   36,   15,
           16,   17,   18,   26,  171,   10,    4,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,   83,   16,  151,   18,  102,  594,  154, 1067, 1068,  256,  594,
          466,  261,   16,   83,   16,  472,   18,   18,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 546,  530,  414, 2079,  307,   13,   92,   16,   29,   18,   26,   13,
           92,   16,   17,   18,   26,  721,   26,   13,   92,   16,  131,   18,
           26,   79,   92,   16,   29,   18,   26,   79,   92,   16,   17,   18,
           26,  721,   26,   79,   92,   16,  131,   18,  131,  307,   11,   91,
           13,   92,   16,   29,   18,   26,   79,   92,   16,   29,   18,   24,
           93,   26,   27,   11,   91,   13,   92,   16,   17,   18,   26,   79,
           92,   16,   17,   18,   24,   93,   26,  721,   26,   11,   91,   13,
           92,   16,  131,   18,   26,   79,   92,   16,  131,   18,   24,   93,
         2080, 2081,  346,  433,  682,  171,  530, 2082,  413,  792,  117, 2083,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   62,   52,   63,   64,   62,   26,    9,   26,   28,   26,   65,
           26,   66,   67,   68,   69,   68,   70,   71,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 125,  861,   91,   17,   34, 1471,   93,   91,  125,   19,   17,   35,
           93,   10,   19,   21,   35,  125,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  145,  151,   51,   56,  142,  242,  151,  238,  261,   16,   83,
           16,  266,   18,   18,  263,   91,  261,   16,   83,   16,  268,   18,
           18,   34,  261,   16,   83,   16,  608,   18,   18,   93,  111,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  135,  136,   65,  137,   13,   15,   16,   17,   18,   19,  116,
           16,   79,   15,   16,   17,   18,   18,   16,   20,   18,   10,   29,
          259,  139,  142,   65,  238,  146,   13,  456,  352,   68,    9,  140,
          990,   29,  584,   20,  991,  145,   83,   16,  809,   18,   53,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 129, 1478, 1479, 1480,  764,   40,  766,  767,  763,  102, 1481, 1482,
         1483,  127,  129,  355,  125,  170,  759, 1484,  130,  125,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   26,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,
           34,   79,  165,   19,   29,  164,   13,   19,   79,  165,   19,   29,
          164,   17,   13,   19,   79,  163,   29,  166,  161,   24,  167,   33,
           10,   13,   34,   17,   79,  168,    4,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1398, 1399, 1400, 1401,  980, 1402, 1403,  662,  980, 1404, 1405, 1406,
         1407,  528, 1408, 1405, 1059,  530,  405,  216,  528, 1408, 1398, 1401,
          528,  682,  201, 1409, 1408, 1401,  528,  682,  426, 1409, 1408, 1398,
         1408, 1401,  528,  682,  351, 1409, 1410, 1408, 1398,  528, 1275, 1410,
          528, 1411,   60,  430,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,   19,
           79,   34,   29,  165,   23,  164,   13,   34,   79,   19,   20,  165,
           23,  164,   13,   19,   20,  163,   23,  166,  161,   24,  167,   33,
           10,   13,   19,   17,   79,  111,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  135,  513,   83,   16,   65,   18,  137,   83,   16,   79,   18,
           15,   16,   17,   18,   10,  134,   83,   16,   13,   18,  139,  464,
           65,  238, 1882, 2066,  361,   79,  949,   68,  465,  464, 1883,  466,
           36, 1883,   36,  179,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  102,  116,   16,  117,   18,   16,   17,   18,  123,  546,   91,
          116,   16,  117,   18,   16,   21,   18,   26,  116,   16,  117,   18,
           16,   17,   18,   93,  251,  888,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  125,
           92,   16,   29,   18,   34,  125,   92,   16,   17,   18,   10,   21,
           26,  125,   92,   16,   20,   18,   19,  125,   92,   16,   29,   18,
           10,  134,  229,   11,   12,  125,   92,   16,  131,   18,   24,   25,
          526,  527,  301,  353,   92,   16,  131,   18,  180,   11,   12,  126,
           92,   16,   20,   18,  125,   92,   16,  131,   18,   24,   25,  181,
           83,   16,  131,   18,  183,  353,   92,   16,  171,   18,   34,  353,
           92,   16,  171,   34,   29,   18,   10,  353,   92,   16,  171,   34,
           20,   18,   26,  229,  171,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   33,   10,   91,  171,   34,   20,   93,   34,   91,  171,   19,
           29,   93,   35, 1273,  242,   68,  645,  646,  861,  171,  254,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 113,   91,   13,   93,   10,   11,   36,   13,   19,  125,   15,   16,
           17,   18,   24,   36,   34,   36,   13,   34,   17,  125,   19,   29,
           36,  125,   10,   17,  249,  229,  290,  113,   91,   13,   93,  165,
           21,  293,  113,   91,   13,   93,  165,   21,   26,  229,  125,  254,
          481,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  180,   12,  125,   92,  131,   25,  125,   92,   16,   29,   18,
           10,   29,   26,   27,   83,   16,  602,   18,   92,   16,  131,   34,
           29,   18,   10,   17,   27,   91,   83,   16,  131,   18,   34,   29,
           93,   27,  125,   92,   16,  131,   18,   26,   27,   61,   83,   16,
          170,   18,   92,   16,  131,   18,   10,  116,   16,   83,   16,  125,
           18,   92,   16,   83,   16,  131,   18,   18,   18,   16,   83,   16,
          131,   18,   18,  229,  170,   92,   16,   29,   18,   26,   27,   83,
           16,  170,   18,   92,   16,   17,   18,   26,   27,   83,   16,  170,
           18,   92,   16,   20,   18,  554,  180,   12,  170,   92,  131,   25,
          132,  449,  229,   12,  125,   92,   16,  131,   18,   25,  526,  527,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   36,   20,   13,   34,   29,
           36,   19,   17,   36,   13,   19,   29,   36,  292,   79,   10,  113,
           91,   13,   93,  120,  229,  290,  113,   91,   13,   93,  257,  113,
           91,   13,   34,   29,   93,  293,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  91,   17,   13,   34,  228,   16,   13,   18,   93,   15,   16,   30,
           18,  582,  102,   83,   16,   13,   18,   15,   16,   20,   18,  583,
            4,  733, 1613, 1784,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  32,   33,   10,   29,   34,   17,   35,   34,   35,   15,   16,   20,
           18,   36,   33,   36,   10,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  226,    9,  584,   28,  584,   65,  125,  584,  170,  584,
          162,    7,  125,  230,    9,   19,  170,  230,   28,   10,   21,  162,
          230,   65,   26,   27,  114,    9,   10,   19,  116,   16,   29,   18,
           16,   21,   18,  116,   16,  170,   18,   16,  162,   18,   10,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,  397, 1630,  829, 1253,  829, 1631, 1632, 1633,   79,   81, 1634,
         1230, 1252,  397, 1635,  829, 1631, 1632, 1633,  284,   79, 1375, 1358,
          279,  138, 1249, 1250,  489, 1630,  829, 1253,  829,   90, 1375, 1358,
          279,  610,  300,  837, 1048,  284,  489, 1245,   16,   79,   18,   10,
           19, 1637,   34, 1638,   83,   16,  279,   18, 1639,  829, 1253,  829,
           90, 1375, 1358,  279,  610,  300,  837,  284,  489, 1245,   16,   79,
           18,   10, 1622,   34, 1640,  279, 1429,  138,  489,  429,  397, 1635,
          829, 1631, 1632, 1633, 1641,  489, 1641,  443, 1642,  449,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  231,  129,  130,  353,  610,  346, 1282,  355,  697,  465,  111,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  513,   65,   79,   15,   16,   17,   18,   10,   21,   13,  139,
          135,  106,  822,  177,  116,   16,   17,   18,   16,   20,   18,  107,
           65,  515,  464,  465,  362,  261,   16,   83,   16, 1882,   18,   18,
          263,  261,   16,   83,   16, 1883,   18,   18,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1725, 1726,  102, 1727, 1728, 1729,  348, 1730, 1729, 1726,    7,  612,
         1731, 1725, 1730,   71, 1732, 1733, 1725, 1730,  880, 1734,  612, 1726,
           71,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 980,   40,  423, 1529,  610, 1530, 1478,  231, 1478,  129,  127,  129,
          355,   13,  131,  300,  355,  125,  130, 1021,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  135,  513,   65,   79,   15,   16,   17,   18,   10,   21,   13,
          139,  173,   79,   10,  116,   16,  278,   18,   16,   13,   18,   91,
          278,  257,   23,   93,   65,  515,   68,  142,  146,  172,  296,   16,
           13,   18,  456,  278,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 370,   11,   12,  125,   92,   16,  131,   18,   24,   25,  102,   26,
           27,  125,   92,   16,   20,   18,   34,  125,   92,   16,   21,   18,
           10,   21,   26,   27,  125,   92,   16,   30,   18,   34,  125,   92,
           16,  133,   18,   10,  105,  229,   11,   12,  125,   92,   16,  131,
           18,   24,   25, 1545,  527,   61,  170,   92,   16,  131,   18,   10,
           11,  118,  125,   92,   16,  131,   18,   24,  119,  229,  180,   11,
           12,  170,   92,   16,  131,   18,   24,   25,  181,   96,  183,   13,
          339,   13, 1546, 1436,   10,   23,   26,   27,  118, 1547,  119,   10,
           17,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    9,   28,   65,   66, 1284,  523,  157,  148,  960,  145,   83,
           16,  151,   18,   56,   53,  366,  228,   16,   20,   18,   26,  246,
           66,  151,  247,  168,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231, 1213, 1214,   52,   51, 1215, 1216,  292,  519,  520,  519,  242,
          800,  801,  802,  877,  519,  247,   91,   27,   93,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 763,  764,   40,  765,  766,  767,  768,  769,  770,  771,  226,  772,
          773,   58,  774,  775,  776,  777,  778,  779,  780,  781,  777,  231,
          777,  782,  232,  783,  777,  782,  784,  785,  786,  777,  782,   50,
          786,  787,  782,  247,  781,  777,  788,    7,  789,  777,  247,  510,
          790,  791,  786,  792,  510,  793,  794,  781,  777,  510,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 371,   99,  372,  373,   92,   16,   29,   18,  137,  220,  220,  374,
          238,  375,  107,  376,  242,  373,   92,   16,   17,   18,  137,  377,
          102,  378,  379,  380,  381,  242,  373,   92,   16,   20,   18,  137,
          377,  359,  107,  374,  359,  107,  382,  373,   92,   16,   21,   18,
          137,  107,  277,  383,  242,  384,   26,  107,  171,  172,  242,  384,
           26,  171,  172,  277,  385,  372,  102,  386,  372,  387,    4,  388,
          373,   92,   16,   29,   18,  389,  373,   92,   16,   21,   18,  390,
          373,   92,   16,   29,   18,  389,  373,   92,   16,   17,   18,  391,
          392,  373,   92,   16,   17,   18,  393,  373,   92,   16,   20,   18,
          394,  392,  373,   92,   16,   20,   18,  393,  392,  373,   92,   16,
           21,   18,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   11,   36,   13,   19,  116,
           16,   29,   18,   16,   17,   18,   24,   36,   34,   11,   36,   13,
           34,  116,   16,   29,   18,   16,   17,   18,   24,   36,  464,  290,
          113,   91,   13,   93, 1574,   17,  293,  229,  464,  241,  125,   26,
          170,  294,  464,  249,   36,  125,   34,  170,   36, 1574,   36,   29,
           34,  575,   36,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231, 1080, 1081,  102,  153,   10,  155,   10,   20,  239,   83,   16,
         1082,   18,   10,  201,   15,   16,   82,   18,  155, 1083,  145,   83,
           16, 1082,   18, 1084, 1085,  464, 1086,   68,   66, 1087,   83,   16,
          153,   18,  172,   83,   16, 1088,   18,  241,  242,  823,  172,  242,
          151,  568, 1089,  594,  238,  142, 1089,  154,  238,   83,   16,  569,
           18,   10,   83,   16, 1090,   18,   10,  116,   16,   17,   18,   16,
           20,   18,   83,   16, 1088,   18,  229,  246,  568,  822, 1091,  247,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  230,   13,   34,  116,   16,   29,
           18,   16,  230,   13,   18,   99,  372,  113,   91,   13,   93,  120,
           79,  735,  113,   91,   13,   93,  120,  141,  736,  113,   91,   13,
           93,  120,  107,   13,   10,  116,   16,  117,   18,   16,   17,   18,
          736,  113,   91,   13,   93,  111,  386,  372,  387,  737,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 581,   11,   91,   13,   34,  116,   16,   79,   15,   16,   17,   18,
           18,   16,   13,   18,   24,   93,   91,   13,   34,   79,   93,   15,
           16,   30,   18,  582,  102,   13,   15,   16,   20,   18,   79,   15,
           16,   20,   18,  583,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1548, 1549,  125,   81,  202,  918, 1548, 1550,  662, 1551, 1551, 1552,
         1549,  825, 1553, 1554, 1555,  530,  405, 1548,  409,  705, 1551, 1050,
         1553, 1059, 1556,  301,    9, 1455, 1551, 1552, 1549, 1557, 1549,  229,
          142,    9,  430,  301,   28, 1455, 1551, 1552, 1549, 1557, 1549, 1557,
         1549, 1558,  229,  142,   28,  430,  229, 1551, 1552,  544, 1549,  430,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  290, 1418,   11,   12,  160,  161,   16,  277,   18,   13,   34,
           79,  167,  167,  105,  164,   17,   13,   19,   79,  165,   23,  166,
          161,   24,  167,  242, 1419,   66,  372,  373,  137, 1420,   91,   13,
           26,   79,   93,  294,   66,   26,   17,   13,   34,   79,  167,  167,
          366,  372, 1421,   91,   13,   26,   79,   93,  294,   66,   26,   17,
           13,   34,   79,   26,  298,  446,   99,  372,  373,  393, 1422,  392,
          373,  393, 1422,  373,  389,  392, 1422,  392,  373,  389,  392, 1422,
           99,  372,  102,  386,  372, 1148,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   91,   13,   19,   17,   93,
          248,   15,   16,   13,   34,  125,   18,   91,   13,   19,   29,   93,
           15,   16,   17,   18,  138,  253,  229,  125,  254,   61,   83,   16,
           13,   18,   92,   16,   29,   18,   26,   27,   83,   16,   13,   18,
           92,   16,   17,   18,   83,   16,  113,   18,   91,   83,   16,   13,
           18,   93,  138,  253,  241,   83,   16,   13,   18,   92,   16,   29,
           18,   34,   83,   16,   13,   18,   92,   16,   17,   18,   22,   17,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1114,  969, 1115, 1116,   96,  417,  445, 1117, 1118, 1119, 1120,  869,
         1121,  734, 1122, 1122,  670, 1121, 1123,  271,  272,  273,  178,  274,
          275,  276,  102,  173,   83,   16,   65,   18,   92,   16,   29,   18,
          178,   79,   10,  278,   36,   13,   36,   34,   17,  140,  141,  280,
           13,  281,  282,  283,  284,  285,  272,  173,   83,   16,   65,   18,
           92,   16,   17,   18,  286,  178,  287,   15,   16,   17,   18,   34,
           17,  287,  114,  288,   19,   20,   10,   23,  229,   83,   16,   65,
           18,   92,   16,   17,   18,  274,  178,   83,   16,   65,   18,   92,
           16,   29,   18,   83,   16,   65,   18,   92,   16,   17,   18,  380,
          381,  149,  289,   68,  229,   83,   16,   65,   18,   92,   16,   29,
           18,  178,    0,    0,    0,    0,    0,    0],
        [ 132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  102,  125,
           92,   16,   29,   18,   10,   29,   26,   27,  125,   92,   16,   30,
           18,   10,   21,  125,   92,   16,   20,   18,  229,   11,   12,  125,
           92,   16,  131,   18,   24,   25,  526,  527,  301,   83,   16,  353,
           18,   92,   16,  131,   18,   11,   12,  125,   92,   16,  131,   18,
           24,   25,  181,  131,  183,  353,   92,   16,  171,   18, 1360,  229,
          171,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 824, 1765, 1766, 1766,  596, 1767, 1768,  203,  596,   98, 1086, 1769,
         1770, 1771, 1765, 1086, 1769, 1770, 1772,  530, 1773, 1375,  339,   96,
         1774,   71,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   62,  957,  669,  151,  238,  150,   62,  145,   83,   16,  151,
           18,   57,  154,  361,  464,  465,  362,   57,   58,  594,  515,   68,
          260,  153,  155,  357,  256,  135,  362,  241,  564,  717,  154,  262,
           54,  150,   62,  523,  603,   10,  470,   10,   17,  228,   16,   20,
           18,  229,  811, 1512,   53,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   16,   13,   79,   36, 1760, 2129,  859,   13,  758,
          295,   79,  758,  295,    9,  102,  484,  307,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  230,  384,   19,  114,  384,   10,  116,   16,   21,   18,   16,
           20,   18,   26,   27,  230,   17,  384,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  135,   92,   16,   29,   18,   26,  135,   92,   16,   17,   18,
          136,   65,  137,   13,   15,   16,   17,   18,   19,  116,   16,   79,
           15,   16,   17,   18,   18,   16,   20,   18,   10,   29,  138,  139,
           62,  140,  141,   68,  142,   65,  143,   36,  144,   36,   10,   17,
          145,  146,   92,   16,   29,   18,  135,   92,   16,   17,   18,   53,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 285,  272,  102,   62,  280,   68,  464,   11,   91,  287,   92,   16,
           23,   18,   26,  288,   92,   16,   23,   18,   24,   93,   11,   91,
          287,   92,   16,   23,   18,  257,   23,   24,   93,  173,   65,  137,
          287,   10,   21,  230,  288,  107,  277,  106,    9,   91,   21,   26,
           23,   93,  848,  352, 1571,  142,  288,   92,   16,   23,   18,   10,
          116,   16,  117,   18,   16,   20,   18,  249,  229,  287,   92,   16,
           23,   18,  277,  286,  178,  464,   65, 1182,  142, 1089,  848, 1721,
          229,  142,   68,  512,  286,  178,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  180,   11,   12,  125,   92,   16,  131,   18,   24,   25,  181,
          131,  183,  353,   92,   16,  131,   18,  125,   92,   16,   29,   18,
           10,   19,   29,  125,   92,   16,  131,   34,   29,   18,   10,  353,
           92,   16,  131,   34,   29,   18,  353,   92,   16,  131,   18,  353,
           92,   16,  131,   18,   10,  179,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   33,   10,   19,   20,   34,   17,   35, 1273,  242,  425,   16,
           33,   18,   68, 1274,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 528, 1361, 1427, 1428, 1429,   20,   66, 1430,  696, 1431,  489,  231,
          489,  750,   63,   19,    9,   92,   16,   29,   18,  154,   92,   16,
           29,   18,   66,   92,   16,   29,   18, 1432,   49,   47,   62,   19,
         1433,  877,  519,   62,  750,   64,  256,   26,  135,   26,  260,   26,
          952,  558,  466,  153,   10,  154,   10,  105,   83,   16, 1019,   18,
           26,  468,   92,   16,   29,   18,   10,   21,   83,   16, 1019,   18,
           20,   66, 1430, 1434,  205, 1435, 1436,  477,  469, 1019,   15,   16,
           20,   18, 1430, 1437, 1431,  489,  828,  205, 1438,    4,  477,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 242,  811,   63,  102,  239,   83,   16, 1903,   18,   10,  201,   15,
           16,   82,   18,  239,    9,   10,  703,   15,   16,   82,   18,  153,
          813,  229,  114,  239,   83,   16, 1904,   18, 1009,   17,  228,   16,
           17,   18,  229,  154,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 764, 1668, 1669,  696, 1670,  217, 1671, 1672, 1668, 1673,  102,   96,
         1674, 1675, 1676,   85, 1674, 1675, 1677,   96, 1674, 1675, 1678, 1671,
         1679, 1668, 1673, 1674,  544, 1675,  430,  179,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  811,   63,  812,  239,  151,   10,  819,   15,   16,   82,   18,
          256,  135,  242,   63, 1743,  362,  814, 1744,  242,   63, 1570, 1744,
          242,   63,  814,   10,   17, 1570,  603, 1744,  821,  241,  242,  816,
         1744,  242, 1745,  229,  107,  603,  107, 1560,  719,  609,  610,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,  113,   91,   13,   93,   10,   17,  114,   13,   34,  230,   13,
          168,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   32,   33,   36,   33,   19,   35,   36,   10,   29,   33, 1273,
          242,   68,   13,   79,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   91,   13,  294,  573,   93,  113,
           91,   13,   93,   10,  113,   91,   17,   19,   13,   93,  112,   79,
           10,   11,   36,   13,   15,   16,   17,   18,   19,   17,   13,   19,
           20,   24,   36,   79,   10,  113,   91,   13,   93,  797,  267,   11,
           91,  160,  161,   16, 1079,   18,   13,   92,   16,   29,   18,   26,
          744,   79,   92,   16,   29,   18,  166,  161,   24,   93,   26,   27,
           11,   91,  160,  161,   16, 1079,   18,   13,   92,   16,   17,   18,
           26,  744,   79,   92,   16,   17,   18,  166,  161,   24,   93,   26,
          721,   26,   11,   91,   13,   92,   16,  171,   18,   26,   79,   92,
           16,  171,   18,   24,   93,  424,   92,   16,   35,   10,   29,   18,
           15,   16,  117,   18,   13,   92,   16,    0],
        [   7,    8,    9,   10,   12,   13,   36,   19,   29,   22,   13,   22,
           17,   25,   26,   27,   28,   10,   12,   13,   36,   23,   22,   13,
           22,   20,   25,   26,    9,  644,   28,   10,   91,   27,   93,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  112,  113,   91,   13,   93,   10,  114,   11,   91,  115,   13,
           34,  116,   16,  117,   18,   16,  105,   18,   24,   93,  118,   19,
          117,   26,  117,  119,  120,  121,  113,   91,   13,   93,  122,  580,
          123,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7, 1841,  173,   65,  137,  116,   16,   13,   15,   16,   17,   18,
           18,   16,  125,   15,   16,   17,   18,   18,   19,  116,   16,   79,
           15,   16,   17,   18,   18,   16,  170,   15,   16,   17,   18,   18,
           10,   29,   27,   91,  125,  257,   23,   26,   27,  170,  257,   23,
           93,  259,  233,    9,   26,   27,    9,  104, 1842,  170,  523,  959,
            9,  104,    9,  136,   65,  175,  360,  361,  464,  465,  362,  239,
           83,   16, 1843,   18,   10,  351,   15,   16,   82,   18,   83,   16,
           65,   18,  457,  737,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  384,  294,   91,   23,   26,  117,   93,   20,  114,  384,   19,
          134,  114,  384,   10,   30,   26,  230,  384,   10,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   32,   33,   33,   34,   35,   10,   20,  822,   35,  425,   16,
           33,   18,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 104,  255,  116,   16,   13,   15,   17,   18,   16,  182,   18,   34,
          116,   16,   79,   15,   17,   18,   16,   21,   18,   10,   29,  149,
          233,  234,   13,  456,  580,  282,  104,  192,  178, 1739,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   36,   17,   13,   19,  125,
           36,   34,  125,  125,  249,  229,  290,  113,   91,   13,   93,  578,
          105,  293,   61,  112,  477,   91,   13,   93,   10,   36,   17,   13,
           19,   29,   36,   13,  758,  573,  249,  113,   91,   13,   93,   34,
          477,   91,   13,   93,  589,   20,  229,  125,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   91,   13,   34,   29,   93,
          174,   13,   19,  125,   91,   13,   19,   29,   93,  125,  249,  229,
          173,   79,   10,  113,   91,   13,   93,   91,   29,  584,  113,   91,
           29,   93,   93,  579,  176,  178, 1566,   13,  294,   91,   29,   26,
           34,  887,   93,  249,  113,   91,   13,   93,  257,   23,  229,  125,
          254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   13,   26,   79,  159,   11,   12,  160,  161,   16,  277,   18,
           17,   13,   34,   20,   79,   19,   20,   22,   23,  164,   17,   13,
           19,   20,   79,   34,   20,  257,   23,  164,   79,   34,   20,  589,
           23,  166,  161,   24,  167,   33,   10,   17,   13,   34,   79,  111,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1412,  151,  226,    9,   28,   65,  125,  170,  162,    7, 1789, 1681,
         1790,  162,  229,   65,   83,   16,  162,   18,   10,  228,   16,  133,
           18,   26,   27,  145,   83,   16,  151,   18,   53,  116,   16,   20,
          228,   16,   20,   18,   18,   16,   17,   18,   26,  229,  145,   83,
           16,  151,   18,  713,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   11,   12,   13,   14,   13,   15,   16,   17,
           18,   19,   20,   13,   19,   21,   22,   23,   24,   25,   26,   27,
           28,   10,   12,   19,   21,   26,   29,   26,   20,   26,   30,   25,
           26,   27,    9,   31,   28,   10,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  513,   65,   79,   15,   16,   17,   18,   10,   17,   13,  139,
          135,  382,   13,  456,  359,  107,  277,   92,   16,   29,   18,  277,
           92,   16,   17,   18,  361,   65,    9,   28,  362,  361,   65,  458,
          142,  568,  362,  135, 1089,  153,  573, 1256,  466,  241, 1257, 1599,
         1258, 1412, 1259,   53, 1412, 1260,   53, 1261,  229,  153,  466,  512,
          178,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 242,  384,  882,   83,   16,   63,   18,   19,   83,   16,    9,   18,
           92,   16,   29,   18,   83,   16,   28,   18,   92,   16,   29,   18,
           83,   16,   65,   18,   92,   16,   29,   18,   83,   16,   66,   18,
           92,   16,   29,   18,  233,    9,  384,  469,  469,  242,   83,   16,
          883,   18,   92,   16,   29,   18,   83,   16,   66,   18,   92,   16,
           29,   18,  384,   31,  242,   63,   10,  171,  384,   31,  242,   83,
           16,  884,   18,   92,   16,   29,   18,   83,   16,    9,   18,   92,
           16,   29,   18,   10,   83,   16,  131,   18,  171,   26,  131,  719,
          720,  741,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 882,   63,   19,    9,   92,   29,   28,   92,   29,   65,   92,   29,
           66,   92,   29,  102,  256,  558,  269,  466,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1965,  406, 1966, 1967,  212,  209,  894, 1968, 1969, 1968,  443, 1970,
         1444, 1444, 1971, 1972, 1367, 1973, 1974, 1975, 1444, 1973, 1974,  530,
          969, 1976, 1977, 1978, 1979, 1977, 1980, 1972, 1965,  712, 1972, 1444,
          447, 1981, 1982, 1974,  447, 1981, 1982, 1974, 1983, 1984, 1444, 1982,
         1985, 1981,  443,  700,  195, 1986, 1444, 1977, 1978, 1974, 1982, 1977,
         1980, 1974, 1094, 1982, 1978,  417, 1980,  417, 1977, 1980, 1974, 1982,
         1977, 1978, 1974, 1094, 1982, 1980,  417, 1978,  417, 1982, 1094, 1982,
          894, 1987,  417,  212,  209,  894, 1981, 1988,  301, 1393, 1662, 1972,
         1444, 1989, 1990, 1892,  973,  229,  973, 1101, 1798, 1991, 1980, 1444,
          249, 1992,  417,  373,   92,   16,   35,   18,   91,   35,   10,   23,
           26,   29,   26,   94,   26,  134,   93,    0],
        [   7,   11,   12,  125,   92,   16,  131,   18,   24,   25,  795,   20,
          370,  180,   11,   12,  170,   92,   16,  131,   18,   24,   25,  170,
           92,   16,   29,   18,   10,   29,   26,   27,  170,   92,   16,   17,
           18,   10,  116,   16,   29,   18,   16,   20,   18,  125,   92,   16,
          131,   18,  170,   92,   16,  131,   34,   29,   18,   34,  170,   92,
           16,  131,   34,   29,   18,   10,  906,   92,   16,  131,   18,  839,
          229,   11,   12,  125,   92,   16,  131,   18,   24,   25,  526,  527,
          842,  229,   11,   12,  170,   92,   16,  131,   18,   24,   25,  181,
          131,  183,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 107,   13,   34,   79,   34,   17,   10,   23,   13,  456,   79, 1220,
            9,   28,  362,   68,  142,  104,   91,   13,   19,   17,   93,   15,
           16,   17,   18,   34,   79,   15,   16,   17,   18,   10,   17,  145,
           83,   16, 1091,   18,   53,  254,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1801,   13,   15,   79,   15, 1762,  483,  234,    9,  107,  277,  106,
           28,   13,  456,  455,  277, 1802,    9,   65,   66,  362,   28,  585,
          155, 1803,  361,  594,  853,  256,  241,   36, 1804,   36,   36,  472,
           36, 1185, 1269,   68,  256,  512,  178, 1270,  256,  512,  173,   65,
           92,   16,   29,   18,   26,  107, 1805,  361,   65,   92,   16,   29,
           18,  464,  465,  362,   28,  502,  352,  107,  104,    9,  361,  142,
          568,  362,  229,  811, 1806,   53,  254,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  290,  291,   61,   13,   26,   79,   26,   33,  294,  573,   13,
           34,   79,   34,   33,   10,   29,  229,   91,   13,   19,   29,   93,
           15,   16,   17,   18,   34,   91,   79,   34,   29,   93,   15,   16,
           17,   18,   34,   91,   33,   34,   29,   93,   15,   16,   17,   18,
          111,   91,   13,   19,   17,   93,   15,   16,   17,   18,   34,   91,
           79,   19,   29,   93,   15,   16,   17,   18,   34,   91,   33,   19,
          125,   93,   15,   16,   17,   18,  165,  116,   16,   29,   18,   16,
           20,   18,  698,  241,  125,  163,   19,   20,  125,  165,   19,   29,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  125,  170,  162,  541, 1165, 1186,  125,   34,  170,   10,  162,
           34,  541,  241,  575,  962, 1514,  228,   16,  125,   18,   34,  228,
           16,  170,   18,  257,  228,   16,  162,   18,   34,  228,   16,  541,
           18,  228,   16,  125,   18,   34,  228,   16,  170,   18,  257,  228,
           16,  162,   18,   34,  228,   16,  541,   18,   36,  125,   19,  170,
           36,   22,   36,  162,   19,  541,   36, 1515,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   12,   19,   29,   26,   23,   26,   29,   26,
           17,   25,   28,   10,   11,   12,   13,   36,   13,   15,   16,   17,
           18,  163,   29,   24,   25,    9,   31,   28,   10,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 61,  13,  79, 159,  11,  12, 160, 161,  16, 277,  18,  17,  13,  34,
          20,  79,  19,  20, 163,  23, 164,  17,  13,  19,  20,  79,  34,  20,
         165,  23, 164,  79,  34,  20, 589,  23, 166, 161,  24, 167,  33,  10,
          17,  13,  34,  79, 111,  91,  27,  93,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [738,   7, 112, 113,  91,  13,  93,  10,  17,  13,  15,  16,  20,  18,
          19, 641,  15,  16,  17,  18,  34, 170, 250, 113,  91,  13,  93, 251,
         252, 125,  26, 170, 113,  91,  13,  93, 546, 111, 168, 229, 794, 125,
          26, 170, 610, 449,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [271, 272, 273, 178, 274, 275, 276, 102, 173,  65,  92,  16,  29,  18,
         178,  79,  10, 278,  36,  13,  36,  34,  17, 140, 141, 280,  13, 281,
         282, 283, 284, 285, 272, 173,  65,  92,  16,  17,  18, 286, 178, 287,
          15,  16,  17,  18,  34,  17, 287, 114, 288,  19,  20,  10,  23, 229,
          65,  92,  16,  17,  18, 274, 178,  65,  92,  16,  29,  18,  65,  92,
          16,  17,  18, 380, 381, 149, 289,  68, 229,  65,  92,  16,  29,  18,
         178,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7, 125,  10,  17,  15,  16, 116,  16,  21,  18,  16,  20,  18,  18,
         170,  10,  20,  15,  16, 116,  16,  17,  18,  16,  20,  18,  18, 162,
          10, 214,  15,  16, 116,  16,  29,  18,  16,  20,  18,  18,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,  750,   83,   16,   63,   18,   19,   83,   16,    9,   18,   92,
           16,   29,   18,   83,   16,   28,   18,   92,   16,   29,   18,   83,
           16,   65,   18,   92,   16,   29,   18,   83,   16,   66,   18,   92,
           16,   29,   18,  102,  153,   10,  182,  154,   10,   96,   83,   16,
          468,   18,   92,   16,   29,   18,   10,  134,   68,  256,  584,  135,
            9,   92,   16,   29,   18,   28,   92,   16,   29,   18,   66,   92,
           16,   29,   18,   65,   92,   16,   29,   18,    9,   92,   16,   29,
           18,  256,   10,   66,   92,   16,   29,   18,  135,   10,   21,  106,
          256,  584,  135,  242,  384,  750,  718,  374, 1504,  858,   52,  121,
          964,  794,   52, 1506,  449,  229,  107, 1162,  242,  384,  719,  720,
          610,    0,    0,    0,    0,    0,    0,    0],
        [ 501,  368,   17,   15,   16,  125,   18,   34,  126,   92,   16,   17,
           18,  125,   10,   21,   15,   16,  170,   18,   34,   17,  126,   92,
           16,   21,   18,  170,   26,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  214,   18,   34,  116,   16,   79,   15,   16,   17,   18,   18,
           16,  171,   15,   16,   17,   18,   18,   10,   29,   91,   23,   22,
          171,   22,   30,   93,  457,  116,   16,  228,   16,  483,   18,   18,
           16,   21,   18,    9,   28,   65,  258,  259,  233,  229,   65,  178,
          567,  142,   65,   68,  568,  107,   13,   10,  105,   36,  569,   36,
           10,   36,  570,   14,   26,   27,  569,  172,  570,  229,  145,  571,
           53,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 965,  447,  966,  967,  968,  969,  131,  667,  970,  971,  131,  667,
          972,   81,  898,   83,   16,  973,   18,   92,   16,   29,   18,   26,
           27,   83,   16,  973,   18,   92,   16,   17,   18,   26,   27,  721,
           26,   27,   83,   16,  973,   18,   92,   16,  131,   18,  446,  702,
          102,  965,  966,  972,  974,  975,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,   65,   92,   16,   29,   18,   65,   92,   16,   17,   18,  273,
          178,   65,   92,   16,   29,   18,   11,   12,  160,  161,   16,  277,
           18,   13,   10,   21,  114,   15,   16,   17,   18,  288,  164,   79,
           10,   21,  230,   15,   16,   17,   18,  288,  166,  161,   24,  167,
          288,  273,   65,   92,   16,   17,   18,   11,   12,  160,  161,   16,
          277,   18,   13,   10,  279,   34,  116,   16,   29,   18,   16,  279,
           18,  164,   79,   10,  279,   19,  116,   16,   29,   18,   16,  279,
           18,  166,  161,   24,  167,  279,  273,   65,   92,   16,   29,   18,
           65,   92,   16,   17,   18,  273,  178,  480,  178,  140,  141,  280,
           13,  281,  282,  283,  284,  285,  272,   61,   65,   92,   16,   29,
           18,   65,   92,   16,   17,   18,  267,    0],
        [  13,   26,   27,   79,  159,   11,   12,  160,  161,   16,  277,   18,
           13,   34,   79,   19,   30,  578,   23,  164,   17,   13,   19,   79,
           19,   29,  589,   23,   26,   27,  478,   16, 1487,   18,   33,   10,
           17,   13,   34,   79,  478,   16, 1488,   18,  164,   13,   19,   17,
           79,   34,   29,  578,   23,  166,  161,   24,  167,    4,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  136,   65,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           91,  125,  257,   23,   26,  170,  257,   23,   93,  258,  259,  139,
          135,   92,   16,   29,   18,   26,  135,   92,   16,   17,   18,  135,
           92,   16,   29,   18,  107,   65,  359,  360,  361,    9,   26,   28,
          362,  261,   16,  135,   92,   16,   29,   18,    9,   18,   10,  261,
           16,  153,   18,   26,  261,   16,  135,   92,   16,   29,   18,   28,
           18,  263,  261,   16,  135,   92,   16,   17,   18,   28,   18,   10,
           23,   65,  457,  179,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,   19,   13,   15,   16,   21,   18,   34,   13,   15,
           16,   17,   18,   34,   17,  120,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,   83,   16,  135,   18,   92,   16,   29,   18,   26,   83,   16,
          135,   18,   92,   16,   17,   18,  136,   83,   16,   65,   18,  137,
          116,   16,   83,   16,   13,   18,   15,   16,   17,   18,   18,   16,
           83,   16,  125,   18,   15,   16,   17,   18,   18,   19,  116,   16,
           83,   16,   79,   18,   15,   16,   17,   18,   18,   16,   83,   16,
          170,   18,   15,   16,   17,   18,   18,   10,   29,   91,   83,   16,
          125,   18,  257,   23,  167,   83,   16,  170,   18,  257,   23,   93,
          258,  259,  139,   62,  140,  141,   83,   16,  135,   18,   92,   16,
           17,   18,  585,   83,   16,   65,   18,  175,  360, 1221, 1571,   83,
           16,  142,   18,   26,   36,  146,   92,   16,   29,   18,   36,   10,
          228,   16,  105,   18,   36,   83,   16,    0],
        [  61,   62,  140,  141,  265,  464,  255,   65,  464,   11,   91,   13,
           92,   16,   23,   18,   26,   79,   92,   16,   23,   18,   24,   93,
          464,  585,   13,  456, 1221, 1571,  465,   68,  142,  261,   16,   83,
           16, 1615,   18,   18,   10,  228,   16,   17,   18,  261,   16,   83,
           16, 1616,   18,   18, 1180,  142,  512,  178, 1270,  568,  107,   13,
          822,  261,   16,   83,   16,  144,   18,   18,  263,  261,   16,   83,
           16, 1256,   18,   18,   10,   29,  241,  106,  142,  591, 1617,  107,
          277,   65,  258,  139,  135,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  126,   92,   16,   17,   18,
           11,   91,   13,   15,   16,   17,   18,   34,  125,   24,   93,  113,
           91,   20,   93,   10,   29,  125,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  226,    9,   28,   65,  125,  170,  162,    7,   65,   10,
          351,   15,   16,   82,   18,   26,  170,   10,  228,   16,  105,   18,
           26,  162,   10,   20,    9,  179,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   10,   12,   19,   29,   26,   23,   26,   29,   26,
           17,   25,   28,   10,   11,   12,   13,   36,   13,   15,   16,   17,
           18,  163,   29,   24,   25,    9,   31,   28,   10,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 882,  242,  801,  802,  631,  520,  231,  801,  247,  631,  247,   60,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  384,  294,   11,   91,   23,   26,  116,   16,  117,   18,   16,
           17,   18,   24,   93,   26,  521,  384,   10,   17,  114,   11,   91,
          384,   19,  116,   16,  117,   18,   16,   21,   18,   24,   93,  179,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   62,  140,  141,  135,  255,   65,  137,  116,   16,   13,   15,
           16,   17,   18,   18,   16,  125,   15,   16,   17,   18,   18,   34,
          116,   16,   79,   15,   16,   17,   18,   18,   16,  170,   15,   16,
           17,   18,   18,   10,   29,   27,   91,  125,  257,  170,  257,   23,
           93,  258,  139,    9,   28,   65,  258,  259,  233,  142,   65,  238,
          146,  172,   13,  456,  106,    9,  107,  277, 1089,  146,  515,   68,
          464,   79, 1220,   68,  256,  107, 1047,  517,  466,   65,  457,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  135,  136,   65,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,
           16,   17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,
           29,   27,   91,  125,  257,   23,   26,   27,  170,  257,   23,   93,
          259,  139,   62,  140,  141, 1416,  604,  104,  104,   13,   15,   16,
           17,   18,   34,   79,   15,   16,   17,   18,   10,  125,   15,   16,
           17,   18,  515,  142,  568,  362,   36, 1256,   36,   36, 1416,   36,
           65,  457,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  113,   91,   13,   93, 1423,  573, 1222,   91,   23,   26,   34,
          887,   93,  251,  799,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  463,   63,  242, 1383, 1384, 1245,   16,   83,   16,  269,   18,
           18,  242,  352,  464, 1245,   16,   83,   16,  269,   18,   18, 1385,
           65,   66,   68,  241,  242, 1386,  815,  242, 1387, 1089,  467,   68,
          142,   83,   16, 1388,   18,  469,  469,  242, 1389,  449,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  102,   28,   10,  116,   16,  117,   18,   16,   21,   18,
          154, 1067, 1212,  116,   16,   29,   18,   16,   20,   18,  154,  230,
            9,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   13,   19,   29,   19,  885,
           16,  125, 2019,   18,   13,  113,   91,   13,   93,  589,   23,   26,
          229,  125,  610,   61,  171, 1546,  378,  697,  131,   26,   11,   91,
           29,   34,  116,   16,   29,   18,   16,   17,   18,   24,   93,   11,
           91,   29,   34,  116,   16,   29,   18,   16,   17,   15,   16,   17,
           18,   18,   24,   93,  721,   11,   91,   29,   34,  116,   16,   29,
           18,   16,   17,   15,   16,  131,   18,   18,   24,   93,   22,  171,
           26,  229,  171,  111,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   83,   16,  135,   18,   92,   16,   29,   18,   26,   83,   16,
          135,   18,   92,   16,   17,   18,  255,   65,  137,  116,   16,   13,
           15,   16,   17,   18,   18,   16,  125,   15,   16,   17,   18,   18,
           34,  116,   16,   79,   15,   16,   17,   18,   18,   16,  170,   15,
           16,   17,   18,   18,   10,   29,   27,   91,  125,  257,  170,  257,
           23,   93,  258,  259,  139,    9,   65,  258,  233,   68,  142,    9,
          177,  116,   16,  228,   16,   20,   18,   18,   16,  105,   18,  107,
          145,   83,   16,  146,   18,   92,   16,   29,   18,   83,   16,  135,
           18,   92,   16,   17,   18,  957,  669,  239,   83,   16,  135,   18,
           92,   16,   29,   18,   83,   16,  135,   18,   92,   16,   17,   18,
           83,   16,  142,   18,   10,  819,   15,    0],
        [ 113,   91,   13,   93,   10,   11,   36,   13,   19,  125,   15,   16,
           17,   18,   24,   36,   34,   36,   13,   34,   17,  125,   19,   29,
           36,  125,   10,   17,  249,  229,  290,  113,   91,   13,   93,  165,
           21,  293,  113,   91,   13,   93,  165,   21,   26,  229,  125,  254,
          481,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  641,   15,   20,   19,   17,
           13,  797,  106,   91,   19,   29,  584,   21,   93,  125,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[106,  91,  17,  26,  29,  93, 104, 220, 356, 357, 234, 107,  17,  13,
          19,  79,  19,  20,  10,  23, 358,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [112,  79,  10, 230,  13,  19, 228,  16,  20,  18, 114,  13, 797, 112,
          79,  10,  17, 230,  13, 797, 348, 259, 876,   4,  81, 110,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7, 169, 889,  16,  83,  16, 125,  18,  18,  10,  91,  29,  26, 171,
          93, 889,  16, 170,  18,  10,  91,  20,  26,  19,  17,  93,  91, 889,
          16,  83,  16, 125,  18,  18,  34, 889,  16, 170,  18,  93, 172, 889,
          16, 170,  18, 171,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7, 125, 257,  23,  26,  27, 170, 257,  23,  26,  27, 125,  15,  16,
          20,  18,  34, 170,  15,  16,  20,  18,  10,  17, 167,  27, 241,  91,
         125,  34, 170,  93,  11,  91, 125,  15,  16,  30,  18,  34, 170,  15,
          16,  30,  18,  24,  93, 589,  21, 125,  34, 170, 163,  17,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7, 1158,  169,  296,   16,  125,   18,   26,  296,   16,  170,   18,
           36,  125,   36,   10,   17,   36,  170,   36,   91,  296,   16,  125,
           18,   19,  296,   16,  170,   18,   93,  172,  296,   16,  170,   18,
          296,   16,  125,   18,  296,   16,  170,   18,  350,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   17,  114,   15,   16,   17,
           18,   13,   19,  230,   15,   16,   17,   18,   13,   34,   17,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  397, 1630,  829, 1253,  829, 1631, 1632, 1633,   79,   81, 1634,
         1230, 1252,  397, 1635,  829, 1631, 1632, 1633,  284,   79, 1375, 1358,
          279,  138, 1249, 1250,  489, 1630,  829, 1253,  829,   90, 1375, 1358,
          279,  610,  300,  837, 1048,  284,  489, 1636,   16,   79,   18,   10,
           19, 1637,   34, 1638,  279, 1639,  829, 1253,  829,   90, 1375, 1358,
          279,  610,  300,  837,  284,  489, 1636,   16,   79,   18,   10, 1622,
           34, 1640,  279, 1429,  138,  489,  429,  397, 1635,  829, 1631, 1632,
         1633, 1641,  489, 1641,  443, 1642,  449,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  519,  520,  149,  523,  878,  879,  359,  880,  352,  523,  804,
          519,  247,  116,   16,  881,  117,   18,   16,   20,   18,  158,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   10,   12,   29,  584,   17,  584,   20,  584,   21,
           25,  584,   28,   10,   12,   17,  584,   21,  584,  105,  584,  134,
           25,    9,   31,   28,  102,  484,  307,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 862,  863,  864,  865,  866,  867,  868,  102,  869,  870,  871,  102,
          872,  870,  871,  102,  873,  874,  868,  871,   71,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   10,   91,   17,   26,   17,   93,
           26,  889,   16,  170,   18,   10,   91,   19,  134,   26,  105,   93,
          114,   22,  889,   16,  125,   18,   26,  889,   16,  170,   18,  257,
           10,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   11,   12,  160,  161,   16,
         1079,   18,  248,   15,   16,   13,   18,   26,  744,   13,  578,   23,
          164,  174,   13,   26,  744,   13,  257,   23,  166,  161,   24,  167,
          477,   91,   13,   93,   10,  113,   91,   13,   93,   34,   13,   34,
          125,  477,   13,  253,  125,  254,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  7,  68, 464, 513,  65,  79,  15,  16,  17,  18,  10,  21,  83,  16,
          13,  18,  26,  65, 139, 177, 278, 107,  65, 515,   9,  28, 362, 239,
          83,  16, 999,  18,  10, 201,  15,  16,  82,  18, 278,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 91,  29,  34,  35,  93,  27,  91,  17,  19,  35,  93,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7, 169, 889,  16, 125,  18,  10,  91,  19,  17,  26,  20,  93,  26,
         889,  16, 170,  18,  10,  91,  20,  26, 171,  93, 889,  16, 125,  18,
         172, 889,  16, 170,  18, 171, 179,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7, 135, 136,  65, 137, 116,  16,  13,  15,  16,  17,  18,  18,  16,
         125,  15,  16,  17,  18,  18,  19, 116,  16,  79,  15,  16,  17,  18,
          18,  16, 170,  15,  16,  17,  18,  18,  10,  29,  91, 125, 257,  23,
          26, 170, 257,  23,  93, 259, 139,   9,  65, 259, 233,  28,  65, 590,
         562, 591,  13, 456, 153, 177,  65, 457,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 212,  209,  611,  612,  613,  614,  615,  616,  496,  617,  618,  619,
          614,  181,  620,  621,  614,  138,  622,  623,  624,  614,  625,  623,
          626,  614,  627,  626,  623,  628,  629,  619,  630,  619,  631,  632,
          633,  614,  628,  629,  619,  622,  634,  635,  614,  636,  620,  212,
          209,  614,  220,  623,   61,  624,  614,  635,   71,  116,   16,   29,
           18,   16,   17,   18,  222,  637,  638,   71,  229,  639,  614,   71,
          640,  634,  635,   71,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  800,  232, 1516,  519,  520,  519,  158,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  107,  502,   13,   19,  228,   16,   20,   18,   79,   34,  105,
           10,   23,  104,   13,   15,   16,   17,   18,   34,   79,   15,   16,
           17,   18,   10,  298,  515,    9,   28,  362,    9,   28,  585,  277,
         1221,   13, 1220,   65,   66,  362,   36,  269,   36,  179,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [2139,  829, 1194, 1097, 2140, 2141, 2142, 1211, 2143, 2144,  621, 2145,
         2146, 2147,  764, 2148, 2149, 2145, 2146, 2147, 2150, 2151, 2152, 2142,
         2153, 2150, 2154, 2140, 2141, 2155, 2156, 2157, 2157, 2158, 1194, 2159,
         1097, 2160,   92,   16,   17,   18,   68, 2161, 2162, 2160,   92,   16,
           17,   18,   68, 2163, 1274, 1194, 2164, 2066,   61, 2165, 1438,  464,
           92,   16,   29,   18, 2145, 1438,  464,   92,   16,   17,   18, 1194,
          358,  573, 2160,   92,   16,   17,   18, 2166, 2145,  358,  451, 2167,
         2168, 2169, 2170,  451,  178,  116,   16,  464,   92,   16,   29,   18,
           18,   16,   91,  573,   34,  451,   93,   15,   16,   17,   18,   18,
           34,  116,   16,  464,   92,   16,   17,   18,   18,   16,  451,   15,
           16,   17,   18,   18,   10,   91,  573,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 513,   65,   79,   15,   16,   13,  139,  135,  177,  107,  361,   65,
          853,  464,  464,   13,  456, 1592,  277,   65,  458,   68,  465,  277,
           83,   16,  470,   18,  172,  277,  464,  107, 1593,  358,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  174,   11,   91,   13,   15,   16,
           17,   18,   19,   17,   13,   19,  134,   24,   93,  251,  888,  546,
           91,   27,   93,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 242,  164, 1776,  882,    9,   28,   65,   66,    9,   16,   28,   16,
           65,   16,   66,   16,  233,    9,  164, 1776,  164, 1777,   16,  242,
          164, 1778,   16,  883,   16,  164, 1778,   16,   66,   16,  164, 1776,
          164, 1779,  242,   63,  171,  164, 1776, 1780,  242,  884,   16,  164,
         1778,   16,    9,   16,  164, 1778,   16,  131, 1487,  164, 1778,   16,
          171,  164, 1778,   16,  131, 1781,  164, 1782,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   10,   19,   17,  112,  113,   91,   13,   93,   10,   11,   91,
           13,   15,   16,   17,   18,   34,  641,   19,   29,   24,   93,  248,
           15,   16,   13,   19,   29,   18, 1113,   68,  113,   91,   13,   93,
         2032,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 271,  290,  291,    7,  125,  584,  170,  584,  162, 1186,  574,   10,
           29,  241,  116,   16,   29,   18,   16,  125,   18,   34,  116,   16,
           29,   18,   16,  170,   18,   34,  116,   16,   29,   18,   16,  162,
           18,  163,  125,   15,   16,   17,   18,   34,  170,   15,   16,   17,
           18,   34,  162,   15,   16,   17,   18,   91,  125,   34,  170,   93,
           15,   16,   20,   18,   34,   91,  170,   34,  162,   93,   15,   16,
           20,   18,   34,   91,  162,   34,  125,   93,   15,   16,   20,   18,
          165,  354,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   52,   63,  979,  980,   40,  981,   52,  982,  102,  983,  984,
           52,   64,  985,  986,   52,  530,  987,  238,  988,  989,  983,   71,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  230,   11,   91,   17,   13,   34,
          116,   16,   20,  117,   18,   16,   17,   18,   24,   93,   19,   20,
          114,   13,  111,    4,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 106,  104,  220,  356,  357,  234,  107,   17,   13,   19,   79,   19,
           20,   10,   23,  358,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 274,  275,  276,  102,  173,   65,   92,   16,   29,   18,  273,  178,
           11,   12,  160,  161,   16,  277,   18,   83,   16,   13,   18,   10,
           83,   16,  125, 1513,  279,   18,   27,   91,   83,   16,  279,   18,
          478,   16,  479,   18,   26,   83,   16,  125,   18,  257,   23,   93,
          167,  478,   16, 1807,   18,  164,   83,   16,   79,   18,   10,   29,
           34,   83,   16,  125,  886,  279,   18,  166,  161,   24,  167,  140,
          141,  280,   13,  281,  282,  283,  285,  272,  102,  173,   65,   92,
           16,   17,   18,  287,   10,   21,  114,  288,   65,   92,   16,   29,
           18,  963,  173,   65,   92,   16,   29,   18,  178,  480,  286,  178,
          107,   65,   92,   16,   20,   18,  286,  178,  288,   10,  384,   92,
           16,   23,   18,   26,  384,   92,   16,    0],
        [ 145,   83,   16,  151,   18,  102,   66,  154,   68,  594, 1493,  239,
           83,   16, 1494,   18,   26,   27,   83,   16,  813,   18,   10,   17,
           83,   16, 1009,   18,  229,  116,   16,  230,  239,   83,   16,   28,
           18,   18,   16,  230,  239,   83,   16,   65,   18,   18,  239,   83,
           16, 1494,   18,   10,  351,   15,   16,   82,   18,   26,   27,  229,
          239,   83,   16,   28,   18,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1273,  242,   32,   33,   10,   35,   91,   19,   17,   34,   35,   93,
           68, 1274,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  232,   59,  523,  524,  525,  232,  523,  157,  247,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,  464,   10,   12,   13,   36,   19,   21,   22,   13,   22,
           17,   25,  584,  465,   10,   11,   12,   13,   36,   13,   15,   16,
           17,   18,   19,   13,   19,  105,   22,   23,   24,   25,   26,  464,
           31,  465,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 980, 1320, 1321, 1322, 1323, 1324, 1325, 1324, 1326, 1327, 1328, 1329,
         1330, 1324,  686,  750, 1327, 1331, 1332, 1323, 1324, 1333, 1329, 1323,
         1324, 1334, 1335,  750, 1336, 1337, 1329, 1323, 1324,  692,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738,  363,  151,  226,    9,   26,   28,   26,   65,  125,   26,  170,
           26,  162,    7,  125,  230,  116,   16,    9,   34,   65,   18,   16,
           17,   18,   10,  170,  230,    9,  229,   28,  363,  151, 1460,  669,
          162,   10,   29,  229,  363,  151,   53,  254,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,   17,  230,   17,   13,  797,  258,  876,  116,   16,
          117,   18,   16,  298,   18,   81,  110,  876,  797, 1783,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,  113,   91,   13,   93,   10,   13,   15,   16,   21,   18,   19,
           17,   13,   15,   16,   20,   18,  120,   68,   91,   29,   26,  113,
           91,   29,   93,   93,  579,  176,  178,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,   83,   16,  151,   18,  226,    9,   28,   65,  125,  170,  162,
            7,  230,   91,    9,   34,   65,   93,   10,  134,  230,   15,   16,
           17,   18,  116,   16,   28,   18,   16,   17,   18,  229,  114,   28,
          125,   34,  162,   10,  105,  145,   83,   16,  151,   18,   53,  229,
          170,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  174,   13,   34,  174,   91,
           17,   19,   13,   93,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   49,   47,  142,  822,   63,  102,   55, 1007,   56,  591,   59,
           63,  153,   10,  154,   10,  116,   16,   29,   18,   16,   17,   18,
          594,   26,   27,  239, 1179,   10,  239,  151,   10,  201,   15,   16,
           82,   18,  256, 1008,  466,  241,  107,   83,   16,  855,   18,  469,
          469,  242,  243,   68,  464,  558,  608,  107, 1047,   59,   63, 1074,
         1075,  703,  820,  229,  607,   83,   16,  464,   18,   19,   83,   16,
          153,   18,   19,   83,   16,   66,   18,  609,  610,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[247, 882, 233, 148, 148, 158,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [136, 116,  16,  13,  15,  16,  17,  18,  18,  16, 125,  15,  16,  17,
          18,  18,  19, 116,  16,  79,  15,  16,  17,  18,  18,  16, 366,  18,
          10,  29,  27,  91, 125, 257,  23,  93, 175, 360, 178,  79,  10, 116,
          16,  20,  18,  16,  30,  18,  13, 125, 179,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [107,  79,  10, 566,  34, 170, 173,  79,  10, 174,  13,  34,  17, 176,
         173,  79,  10, 174,  91,  13,  34,  29,  93, 176, 170, 179,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 61,  62, 140, 141, 107,  13,  10, 125, 136, 116,  16,  13,  15,  16,
          17,  18,  18,  16, 125,  15,  16,  17,  18,  18,  19, 116,  16,  79,
          15,  16,  17,  18,  18,  16, 170,  15,  16,  17,  18,  18,  10,  29,
          91, 125, 257,  23,  26, 170, 257,  23,  93, 359, 360, 361,  66,  26,
         256, 362, 363, 364,  53, 134,  26,  65, 365, 111,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  349,  296,   16,  125,   18,  296,   16,  170,   18,  350,  703,
           15,   16,   82,   18,  278,  296,   16,  125,   18,   19,  296,   16,
          170,   18,  296,   16,  125,   18,  352,  278,   10,    4,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  153,  150,   83,   16,   62,   18,  604,  155,  150,   83,   16,
           62,   18,  176,  154,  361,  150,   83,   16,   62,   18,  853,  256,
           66,  155,  466,  241,  560,  150,   83,   16,   62,   18,  176,  854,
           10,  228,   16,   20,   18,  855,  229,  239,  856,  857,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  107,  277, 1312,   34,   79,   34,   20,  171,   19,  228,   16,
           20,   18,   10,   23,  104,   13,   15,   16,   17,   18,   34,   79,
           15,   16,   17,   18,   10,  298,  515,    9,   28,  362,    9,   28,
          585,  277, 1221,   13, 1220,   65,   66,  362,   36,  153,   36,   10,
           17,  228,   16,   20,   18,   36,  269,   36,   10,  179,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  915,   63,  102,  145,  151,  236,  594,   10,  269,  241,  155,
          172,  813,    7,  145,  823,  395,  153,   10,  813,  256,  558,  813,
           66,  455,   68,  603,  172,  821,  229,  915, 1310,  915, 1311,  247,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 72,  73,  74,  75,  76,  77,  78,  79,  80,  13,  81,  15,  16,  82,
          18,  83,  16,  65,  18,  84,  85,  80,  86,  87,  88,  89,  89,  90,
          11,  91,  13,  92,  16,  35,  18,  26,  79,  92,  16,  35,  18,  24,
          93,  91,  35,  10,  29,  26,  17,  26,  94,  26,  85,  93,  95,  95,
          96,  15,  16,  82,  18,  83,  16,  65,  18,  97,  15,  16,  82,  18,
          83,  16,  65,  18,  98,  99, 100, 101, 102, 103,  78,  79,  80,  13,
         100, 101,   3,   3,   3,   3,   3,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [581,  11,  91,  13,  15,  16,  17,  18,  34, 116,  16,  17,  18,  16,
          13,  18,  24,  93,  15,  16, 105,  18, 582, 102, 731, 732,   4, 733,
         734,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [704, 308, 705, 612, 706, 330, 707, 708, 709, 710, 711, 612, 706, 706,
         348, 712, 705, 528, 712, 413, 179,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7, 112, 113,  91,  13,  93,  10,  36,  13,  34,  29,  36,  19,  17,
          36,  13,  19, 125,  36,  26, 125, 257,  23, 125,  10,  29, 249, 229,
         290, 113,  91,  13,  93, 257,  29, 293, 113,  91,  13,  93, 797,  13,
         456, 858,  56,  53, 754, 229, 125, 254,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,   62,  140,  141,  135,  255,   65,  116,   16,   13,   15,   16,
           17,   18,   18,   16,  125,   15,   16,   17,   18,   18,   34,  116,
           16,   79,   15,   16,   17,   18,   18,   16,  170,   15,   16,   17,
           18,   18,   10,   29,   27,   91,  125,  257,  170,  257,   23,   93,
          258,  139,    9,   28,   65,  258,  259,  233,  142,   65,  238,  146,
          172,   13,  456,  106,    9,  107,  277, 1089,  146,  515,   68,  464,
           79, 1220,   68,  256,  107, 1047,  517,  466,   65,  457,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1213, 1214,   52,   51, 1215, 1216,  292,  519,  520,  519,  242,
          800,  801,  802,  877,  519,  247,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  800,   59,   64,   83,   16,   62,   18,   92,   16,   29,   18,
           26,   83,   16,   62,   18,   92,   16,   17,   18,  107,   83,   16,
           62,   18,   92,   16,   29,   18,   83,   16,   62,   18,   92,   16,
           17,   18,  242, 1064,  800,  877, 1065,   53,   52,  800,  158,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1124, 1125, 1126, 1127, 1128, 1129,  110, 1129, 1130,  110,  116,   16,
          228,   16,   30,   18,   19,   29,   18,   16,   17,   18,  116,   16,
          228,   16,   30,   18,   19,   29,   18,   16,   17,   18, 1131,  662,
         1132, 1060, 1133, 1134, 1135, 1136, 1126, 1127, 1128, 1137,  110, 1137,
         1129,  110,  116,   16,  228,   16,   30,   18,   19,   29,   18,   16,
           17,   18, 1138,  138, 1132, 1060, 1139,  525, 1140, 1128, 1141, 1142,
          110, 1143, 1019, 1144,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   16,   28,   10,   12,   13,   36,   91,   13,   34,
           29,   93,   91,   13,   19,   17,   93,   22,   23,   26,   13,  294,
          295,   25,    9,  644,   28,   54,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   13,   15,   16,   20,   18,
           34,   91,  125,   19,   29,   93,   13,   15,   16,   17,   18,   34,
          641,  113,   91,   13,   93, 1066,  173,   79,   10,  113,   91,   13,
           93,   68,  990,   23,  584,   23,  991,  579,  176,  178,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  125,  170,   36,  125, 1647,  170,   36,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  102,   66,  154,   68,  594, 1493,  239,   83,   16, 1494,
           18,  145,   83,   16, 2016,   18,   53,  145,   83,   16, 1903,   18,
           53, 2043,  229,  116,   16,  230,  239,   28,   18,   16,  230,  239,
           65,   18,   83,   16,  594,   18,   10,   29,   83,   16, 1009,   18,
           10,  116,   16,  228,   16,   17,   18,   18,   16,   17,   18,  229,
          813,  155, 1595,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 764, 1884, 2130, 2131,   76,  102, 2132, 2133, 2131,  754,  994,  138,
         2134,  596,   10,  133,   34, 1119,  339,  596, 2134,  102,  530, 1367,
          138,  433,   54,  596,   71,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93,   10,   36,   13,   19,  125,   36,   13,
           34,   36,   13,   19,   17,   36,   91,   13,   19,  125,   93,  125,
           10,   29,  249,  229,  290,  113,   91,   13,   93,   22,   23,  293,
           13,  294,   91,   19,  887,   26,   29,   93,  249,  113,   91,   13,
           93,   22,   23,  229,  125,  254,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301,   83,   16,  353,   18,   92,   16,  131,   18,  370,   11,   12,
          125,   92,   16,  131,   18,   24,   25,  181,  131,  183,    7,  125,
           92,   16,   29,   18,   10,   19,  133,   26,   27,  353,   92,   16,
           20,   18,   10,   19,  483,  229,   11,   12,  125,   92,   16,  131,
           18,   24,   25,  526,  527,  229,   83,   16,  353,   18,   92,   16,
          131,   18,  914,   83,   16,  353,   18,   92,   16,  131,   18,  111,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  248,   15,   16,   17,   13,
           18,   19,  125,  174,   13,  250,  113,   13,  850,  112,  113,  851,
           13,  253,  307,  241,  125,  249,  113,   91,   13,   93,  589,   17,
          125,   34,  125,  174,  116,   16,   17,   18,   16,  125,   18,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  125,  170,  162,  541, 1165, 1186,  125,  170,  162,  541,  241,
          575,  257, 1514,   26,   27,  228,   16,  125,   18,   34,  228,   16,
          170,   18,  257,  228,   16,  162,   18,   34,  228,   16,  541,   18,
          228,   16,  125,   18,   34,  228,   16,  170,   18,  257,  228,   16,
          162,   18,   34,  228,   16,  541,   18,   36,  125,   19,  170,   36,
           22,   36,  162,   19,  541,   36, 1515,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1409, 1897, 1409, 1898, 1899,  330, 1900,  614,  348, 1409, 1897, 1901,
          969, 1902,  179,  733, 1613, 1784,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  83,   16,   13,   18,   26,   79,  159,   11,   12,  160,  161,   16,
          277,   18,   83,   16,   13,   18,   19,   83,   16,   79,   18,  589,
           23,  164,   83,   16,   13,   18,   34,   83,   16,   79,   18,   19,
           17,  578,   23,   26,  478,   16, 1487,   18,   83,   16,   33,   18,
           10,   20,   83,   16,   13,   18,   19,   21,   83,   16,   79,   18,
          478,   16, 2012,   18,  164,   79,  589,   23,  166,  161,   24,  167,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  150,   62,  102, 1245,   16,  153,   18,  466,  142,  109,  608,
         1008,  361,  153,  256,  135,  362,  239, 1262,   10,   17,  239, 1263,
          229,  239, 1263,  857,  821, 1264, 1265, 1266, 1264, 1267,   68,  260,
          241, 1268,  172,  269,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   10,   12,   13,   36,   36,   13,   36,   22,   20,
           26,   13,  294,  295,   25,   26,   27,   28,   10,   12,   13,   36,
           36,   13,   36,  257,   29,   26,   13,  294,  296,   16,  295,   18,
           25,   26,   27,    9,   31,   28,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 528,  529,  530,  405,  531,  216,  532,  533,  194,  534,  535,  536,
          537,  218,   90,  538,   81,  500,  450,   27,  503,   15,   16,   17,
           18,   10,  116,   16,  131,   91,  539,   19,  540,   93,   15,   16,
           17,   18,   18,   16,   91,  125,   34,  170,   93,   91,  162,   34,
          541,   93,   91,  125,   34,  162,   93,   91,  170,   34,  541,   93,
           18,  219,  542,  533,  194,   71,  229,  543,  535,  536,  544,  537,
          430,  486,  102,   90,  545,  546,  547,  548,  549,  533,  194,  303,
          550,  533,  549,  533,  194,  303,  550,  533,  551,   90,  552,  553,
          553,  554,  555,  556,  543,  535,  536,  537,  542,  534,  533,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,   92,   16,   29,   18,  273,  178,
          279,  273,  125,  962,   23,  140,  141,  280,   13,  281,  282,  283,
          285,  272,  102,  173,   83,   16,   65,   18,   92,   16,   17,   18,
          137,   27,  287,   10,   21,  114,  288,  839,   65,   92,   16,   29,
           18,  963,  173,   65,   92,   16,   29,   18,  178,  480,  286,  178,
          842,  107,   65,   92,   16,   20,   18,  286,  178,  288,   10,  384,
           92,   16,   23,   18,  384,   92,   16,   23,   18,  521,  384,   92,
           16,   23,   18,   10,   17,  173,   65,   92,   16,   29,   18,   65,
           92,   16,   17,   18,  289,   68,   65,   92,   16,   20,   18,  229,
          125,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  366,   13,   15,   16,   17,   18,   34,   79,   15,
           16,   17,   18,   10,  171,   15,   16,   17,   18,   91,  171,  257,
           23,   93,  107,  277,  141,   62,  382,  356,  277,   65,  138,  267,
            9,   28, 1089,  153,  466,  464,  241,  107,  848,  177,  277,  177,
         1509, 1185,  277,  106,   11,   91,  116,   16,  171,   18,   16,   20,
           18,   26,  171,   24,   93, 2066, 2067,  848,   65,  515,   68,  142,
          811, 2068, 1080, 2069,  229,  277,  177,  449,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  255,  256,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          279,   18,   34,  116,   16,   79,   15,   16,   17,   18,   18,   16,
           20,   18,   10,   29,  139,   13, 1314,    9,  256,  258,  233,  177,
          278,   91,  278,  962,   23,   93,  107,  361,  256,    9,  464,  362,
           68,  465,  256, 1183,  172, 1567,  279,   10,   21,   36,  467,   36,
           10,   36, 1568,   36,  249,  229,  145,  471,   53,   17,   36,  467,
           36,   10,   36, 1568,   36,  249,  229,  278,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  125,  170,  162,  145,   83,   16,  151,   18,  226,    9,   28,
           65,  230,   15,   16,   17,   18,   83,   16,   28,   18,   10,   17,
          230,   83,   16,    9,   18,  230,   83,   16,   65,   18,  125,  170,
          229,  810,   61,   83,   16,   28,   18,   10,  201,   15,   16,   82,
           18,  125,   10,  228,   16,   17,   18,  229,  145,   83,   16,  151,
           18,   53,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  102,  114,   65,   10,  116,   16,   17,   18,   16,   20,
           18,   26,  155,   10,   21,   26,  154,   10,   20,   26,  521,   28,
           10,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  37,   38,   39,   40,   41,   42,   43,   44,   45,   46,   47,   48,
           49,   47,   50,   51,   52,   53,   54,   49,   47,   55,   56,   53,
           55,   56,   57,   58,   59,   52,   51,   60,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 182,  417,  125,  170,  377,  102,  359,  352,  107, 1809,  395,  151,
         1880,  155,  107,  125,  170,  352, 1890,  153,  107,  155, 2013, 2014,
         1062,  107,  153,  125,  985,  351,  820, 2015,  153,  170,  985,  596,
          820,  647,  107,  153,  125,  985,  351,  820, 2015,  153,  170,  985,
          351,  820,  647,  107,  153,  125,  719,  111,  703,  820,  107,  153,
          125,  719,  111,  351,  820, 1063, 1533, 1613, 1063, 1062, 1148,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 824,  917,  918,   29,  919,  920,  921,  920,  922,  619,  920,  923,
          924,  925,  920,  249,  926,  918,  924,  927,  409,  202,  920,  313,
          927,  918,  333,  202,  928,  929,  918,  920,  930,  918,  931,  923,
          924,  932,  218,  216,  919,  920,  922,  933,  934,  923,  924,  433,
          231,  935,  301,   13,   29,  919,  920,  922,  936,  937,  934,  923,
          924,  433,   79,   29,  919,  920,  918,  923,  924,  828,  938,   81,
          202,  131,  939,  918,  923,  924,  433,  839,  131,   10,  511,  229,
           79,   13,  112,  940,  941,  842,  828,  934,  923,  924,  433,  754,
          131,  942,  943,  900,  229,  131,  111,  944,  216,  919,  920,  939,
          945,  918,  511,  923,  924,  945,  918,   85,  923,  924,  423,  216,
          919,  920,  918,  923,  924,  828,  938,    0],
        [   7,  112,  113,   91,   13,   93,   10,   17,  230,   13,   34,  230,
           17,   13,  113,   91,   13,   93,  111,    4,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   16,   13,   36, 1760, 2138,   28,   16,   13,   36,
           13,    9,  760,   28,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 230,  384,   10,   16,  116,   16,   29,   18,   16,   20,   18,   18,
          114,   17,  125,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 180,   11,   12,  125,   92,   16,  131,   18,   24,   25,  125,   92,
           16,  131,   34,   17,   18,   34,   91,   19,   29,   93,   15,   16,
          131,   18,  125,   92,   16,  131,   18,   10,   20,  131,   19,   29,
          181,  182,  183,  184,   11,   12,  125,   92,   16,   29,   18,   24,
           25,   10,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  852,   62,  102, 1245,   16,  153,   18,  466,  142,  109,  608,
         1008,  361,  153,  256,  135,  362,  818, 1262,  818, 1263,  229,  818,
         1263,  857,  821, 1264, 1265, 1266, 1264, 1267,   68,  260,  241, 1268,
          815,  269,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7, 1223, 1165, 1186,  132,   11,   12,  125,   92,   16,  131,   18,
           24,   25,  181,  183,  483,  125,   92,   16, 1021,   18,   10,   20,
          125,   92,   16,   20,   18,   34,   21,  125,   92,   16,   29,   18,
          125,   92,   16,   20,   18,   10,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,    9,   28,   65,   66, 1284,  523,  157,  148,  960,  145,   83,
           16,  151,   18,   56,   53,  366,  228,   16,   20,   18,  246,   66,
           19,  151,  247,  168,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,   34,
           17,   79,   19,   30,  165,   23,  164,   13,   19,   17,   79,   34,
           20,  165,   23,  164,   13,   19,   30,  163,   23,  166,  161,   24,
          167,   33,   10,   13,   34,   79,  168,  179,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 136,  116,   16,   13,   15,   16,   17,   18,   18,   16,  125,   15,
           16,   17,   18,   18,   19,  116,   16,   79,   15,   16,   17,   18,
           18,   16,  170,   15,   16,   17,   18,   18,   10,   29,   27,   91,
          125,  257,   23,   26,  170,  257,   23,   93,  457,  228,   16,   20,
           18,  360,  178,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  116,   16,  230,   13,   34,   13,
           18,   16,  114,   13,   34,   13,   15,   16,   17,   18,   18,  118,
           19,  117,   26,  117,  119,  120,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 371,   99,  372,   83,   16,  373,   18,   92,   16,   29,   18,  137,
           32,   83,   16,   33,   18,  116,   16,   29,   18,   16,   83,   16,
           33,   18,   18,  294,   83,   16,  573,   18,   26,   83,   16,   33,
           18,  294,   83,   16,  573,   18,   83,   16,  373,   18,   92,   16,
           17,   18,  137,   32,   33,   33,   15,   16,   17,   18,  294,   83,
           16,  573,   18,   26,   92,   16,   83,   16,   33,   18,   18,  294,
           83,   16,  573,   18,   92,   16,  478,   16, 1808,   18,   18,   83,
           16,  373,   18,   92,   16,   20,   18,  137,   32,   83,   16,   33,
           18,   92,   16,   29,   18,   26,   27,   83,   16,   33,   18,   92,
           16,   17,   18,   83,   16,   33,   18,   92,   16,  298,   18,  294,
           83,   16,  256,   18,   26,   27,   83,    0],
        [  61,   62,  140,  141,  107,   13,  513,   65,  726,   79,   15,   16,
           17,   18,   10,   17,  514,   91,  373,  257,   23,   93,  515,   66,
          256,  362,  516,  172,  517,   65,  139,  140,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  145,  151,   53,  116,   16,  366,  228,   16,   20,   18,   18,
           16,   21,   18,   56,  233,  157,   62,  148,  367,   62,  158,  182,
          117,   26,   62,  242,  151,  358,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  764, 1224,  829, 1225,  829, 1226,  707, 1227, 1228,   81, 1229,
         1230, 1231, 1232, 1233,  822, 1232, 1224,  822, 1225, 1234, 1235,   90,
          424,   92,   16,   35,   10,   29,   18,   15,   16,  133,   18,   79,
           92,   16,   35,   18,   10, 1237,   26,   27,  424,   92,   16,   35,
           10,   29,   18,   15,   16,  133,   18,  279,   92,   16,   35,   18,
           79,   92,   16,   35,   18,   10, 1238,   26,   27,  228,   16,  424,
           92,   16,   35,   10,   29,   18,   15,   16,  133,   18,   11,   91,
           79,   92,   16,   35,   18,   19,  425,   16,   79,   18,   24,   93,
           15,   16,   17,   18,   18,   10, 1242,   26,   27,  228,   16,  133,
           18,  452, 1244, 1235,  527,  434,  451,   10,  116,   16,  424,   92,
           16,   35,   10,   29,   18,   15,   16,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,  230,   13,   19,  174,   91,
           29,   34,   13,   93,  584,   27,  113,   15,   16,  522,   18,   91,
           13,   93,  113,   91,   13,   93,  369,  241,  113,   15,   16,  522,
           18,   91,   13,   93,  546,   11,   91,   19,   29,   26,  116,   16,
          117,   18,   16,   17,   18,   24,   93, 1181, 1964,   68,  113,   91,
           13,   93,  380,  381,  253,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93,   10,   36,   13,   34,   29,   36,   19,
           36,  641,   19,   29,   36,  125,   10,   29,  249,  229,  290,  113,
           91,   13,   93,  962,   29,  293,   13,  294,   91,   23,   26,   29,
           93,  249,  290,  113,   91,   13,   93,  257,   13,  698,  229,  125,
          254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 513,   79,   15,   16,   17,   18,   10,   17,  514,   91,  373,  257,
           23,   93,  139,  255,  116,   16,   13,   15,   16,   17,   18,   18,
           16,   20,  373,   18,   34,  116,   16,   79,   15,   16,   17,   18,
           18,   16,  373,   18,   10,   29,  139,  373,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    9,   28,  255,  256,  137,  116,   16,   13,   15,   16,   17,
           18,   18,   16,  125,   15,   16,   17,   18,   18,   34,   79,   15,
           16,   17,   18,   10,   29,   27,  125,  257,   29,  258,  259,  233,
          260,  256,  233,  261,   16,  262,   18,  263,  261,   16,  264,   18,
           10,  134,  142,  107,   13,   10,  105,  265,  266,  256,  267,   65,
          268,  256,  267,   66,  229,  256,  178,  241,  107,  269,  270,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1704, 1705, 1706, 1707,  893, 1059,  530,  405,  216,  892,  892, 1708,
          181, 1050, 1708, 1709, 1710,   79, 1100, 1101, 1102,  219, 1711,  892,
          102, 1709, 1710, 1301, 1712,  892, 1060, 1709, 1713,  892, 1060,  229,
         1714,  892, 1709, 1710,  431,  725,  430,  486,  102,   90,  545,  546,
          547,  548,  435,  436,  450,  228,   16, 1715,   18,  452, 1716,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,   13,  174,   91,   13,   34,  228,
           16,  125,   34,   13,   15,   17,   18,   93, 1222,  125,  905,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 824, 1495,  190, 1496,    9,   28,  220,  397,  530,  405,   97, 1495,
         1495,  190, 1496, 1497,    9,  397, 1495, 1496, 1497,  942, 1101, 1103,
           28,  397, 1495, 1496, 1497, 1100, 1101, 1102,   28,  397, 1495, 1496,
         1497, 1100, 1101, 1102, 1498,   28,  397, 1495, 1496, 1497,  942, 1101,
         1103, 1103,  220,  397, 1496, 1497, 1459, 1499,  975,  423,  794,  610,
          446, 1062,  838, 1495, 1496, 1497, 1495, 1496, 1500,   58,  193,  149,
          194,  219,  397, 1495, 1496,  194, 1173,   71,  449,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  246,  142,  822,  151,  102,  153,  154,   17,  228,   16,   17,
           18,  266,  268,  608,  155,   62,  155,  466,  241,  605,  172,  242,
          151,  567,  464,  558,  154, 1388, 1643, 1180,   65,  242, 1644,  358,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 368,    7,   30,   15,   16,   30,   18,   22,  134,   15,   16,   21,
           18,   26,  729,   15,   16,   21,   18,   22,  134,   15,   16,   30,
           18,   61,  125,   10,  126,   92,   16,   20,   18,   20,   26,   27,
          170,   10,  126,   92,   16,   20,   18,   30,   26,   27,  162,   10,
          126,   92,   16,  729,   18,  134,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 812,   63, 1559,  155,  813,  515,   68,   62,  153,  155,   68,  256,
          135,  594,  269,  603,   10, 1560,   10,  116,   16,   30,   18,   16,
           21,   18,  564,  361,  813,  853,  952,  145, 1561,  564, 1562,  145,
           66,   15,   16,  522,   18,  564, 1087,  516,   15,   16,  522,   18,
           10,  228,   16,   96,   18,  241,   66,   15,   16,  522,   18,  952,
          172,  242,   63,  229,  607,   28,   19,   66,   15,   16,  522,   18,
            9,   19,   65,  720,  610,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 230,   85,   15,   16,   82,   18,  114,   96,   15,   16,   82,   18,
           19,  114, 1723,   15,   16,   82,   18,  230,   96,   15,   16,   82,
           18,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  135,  513,   65,   79,   15,   16,   17,   18,   10,   21,   13,
          139,  135,  585,  359,  352,  107, 1814, 1815,  107, 1814,   65,  515,
            9,   28,  362,  107, 1815,   65,  515,   66,  256,  362,   36,  153,
           36,   36,  560,   36,  111,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1285,  255,   65,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   34,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           27,   91,  125,  257,  170,  257,   23,   93,  258,  259,  233,    9,
           92,   29,   26,    9,   92,   17, 1089,    9,   92,   29,    9,   92,
           17,  604,  104,  107,  752,   19,  890,   34,   17,  575,   10,   23,
          357,   65,  457,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,   79,   10,  113,   91,   13,   93,  797,   79,   10,   17,
           15,   16,   13,   15,   16,   34,   18,  125,   18,  797,   79,   10,
           19,   13,  736,  113,   91,   19,   17,   93,   34,  113,   91,   19,
           21,   93,   10,   29,  125,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 212,  209, 1543, 1576, 1577,  612,  775, 1578, 1579, 1580, 1581, 1582,
         1578, 1541, 1583,  102, 1409, 1584, 1409,  212,  209, 1543, 1582,  209,
         1543, 1582, 1576,  212, 1582, 1585,  212, 1541, 1582, 1586,   91,   27,
           93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,  273,  178,   11,   12,  160,  161,
           16,  277,   18,   13,   10,   17,  114,  288,  164,   79,   10,   21,
          230,  288,  166,  161,   24,  167, 1392,  273,  107,  277,  273,  178,
           11,   12,  160,  161,   16,  277,   18,   13,   10,   29,   34,  279,
          114,  384,  164,   79,   10,   17,   34,  885,   16, 2137,   18,  384,
          166,  161,   24,  167,  279,  273,  229,   65,  277,  274,  178,  173,
           65, 1646,  107,  277,  877, 1089,  466,  140,  229,  277,  177,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,  151,  102,  230,   15,   16,   17,   18,    9,   19,  230,   15,
           16,   17,   18,   28,   19,  230,   15,   16,   17,   18,   65,   10,
          230,   28,  230,   65,  229,    9,  154,   10,   20,  229,  145,  151,
          713,  168,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   16,   28,   16,   13,   36,  859,   13,  859,    9,
          760,   28,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1361,  406,  190, 1912, 1913, 1914,  409,  191, 1915,  190, 1916, 1495,
          190,  585, 1917, 1917,  794, 1918,  934, 1919, 1917,  249, 1920, 1921,
          190,   67,   85,  191,  585, 1917, 1917,  872,  190,  585, 1917,   61,
          223, 1918,   71,  373,   91,   23,   22,  373,   22,   29,   93,  223,
         1918,  880, 1734,  301,   85,  223, 1879,  191, 1918,   71,  113,   91,
          373,   93,  229,  113,   91,  373,   93,  168,   68,  373,   92,   16,
           23,   18, 1922, 1915, 1923,   85,  191, 1652,  191, 1918,  102,  373,
           92,   16,   23,   18,  373,  610,    7,  223, 1917,  938,  202, 1924,
         1918, 1495, 1925, 1361,  198, 1918, 1926,  214,  202,  204,  938,   35,
         1927,  872,  190,  585, 1917, 1915,  190, 1917,  938,  204,  938, 1892,
          973,  229, 1928,  846, 1917,  938,  204,    0],
        [  61,  124,  125,  126,   92,   16,   20,   18,   21,   10,   17,   26,
           21,   15,   16,   19,  125,   18,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 127,  231,  129,  355,  125,   10,   21,  584,  170,   10,  105,  130,
          131,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  125,   26,  170,   36,  125,   36,   10,   30,   26,   36,
          170,   36,   10,  105,   26,  125,  263,  170,   10,   19,  105,  114,
          727,  125,   26,  125,   34,  170,  728,   10,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 173,   79, 1663, 1664,   68,  805, 1647,  579,  176,  178,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  33,   91,   29,   34,   35,   93,   10,   17,   35,   33,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  513,   83,   16,   65,   18,  137,   83,   16,   79,   18,   15,
           16,   17,   18,   10,   21,   83,   16,   13,   18,  139,  135,  135,
          177,  278,   91,  278,  257,   23,   93,  107,  277,   65,  515,    9,
           28,  362,   36,  153,   36,  229,  277,  178, 1645,   68,    9,   28,
           65,  458,  357,  104,  178,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  129,  229,  794,   20,   15,   16,  131,   18,   19,   17,   15,
           16,  131,   18,  257,  993,  122,  994,  131,  812,  463,  138,  995,
          996,  997,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 521, 1157,   15,   16,   82,   18,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  290,  291,    7,  112,  113,   91,   13,   93,   10,   36,   20,
           13,   34,   29,   36,   19,   17,   36,   13,   36,  292,   79,   10,
          113,   91,   13,   93,  120,  229,  290,  113,   91,   13,   93,  257,
          113,   91,   13,   34,   29,   93,  293,    3,    3,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 127,  231,  129,  130,  353,  610,  346, 1282,  355,  697,  465,  111,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1217, 1177,  459,  460,  151,  822,    9,   92,   29,   28,   92,   29,
           65,   92,   29, 1218,  247, 1219,  157,  153,  172,  154,  584,  153,
           10,  105,  584,  154,   10,  134,  584,  468,   92,   29,   10,   20,
         1219,  168,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 212,  209, 1942, 1943, 1944, 1945, 1946, 1947, 1948,  638, 1949,  249,
         1950,  635, 1944,  636, 1951,  614, 1582, 1952, 1953,  712,  300, 1954,
         1955, 1956, 1955, 1954,   61, 1952, 1957, 1958,   71, 1732, 1959, 1958,
           71,  900, 1960, 1961,  880, 1734, 1952, 1962,  635,   71,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 423,  353,   10,   29,   19,  116,   16,   29,   18,   16,   17,   18,
           34,  116,   16,   29,   18,   16,   20,   18,   19,  116,   16,   29,
           18,   16,   21,   18,   34,  721,   34,  116,   16,   29,   18,   16,
         1622,   18,   19,  116,   16,   29,   18,   16,  216,   18, 1623,  231,
          129,  995, 1159,  997,   91,   27,   93,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  13,   26,   27,   79, 1628,   86,   11,   12,  160,  161,   16,  277,
           18,   13,   34,   17,   79,   19,   30,  589,   23,  164,   13,   19,
           17,   79,   34,   20,  165,   23,  164,   13,   19,   30,  163,   23,
          166,  161,   24,  167,   33,   10,   13,   34,   79,  168,    4,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  173,   79,   13, 1490,   68,  579,  176,  173,   79,   10,  641,
           15,   16,   17,   18,   34,   91,  125,   34,   17,   93,   13,   34,
           29, 1491, 1492,  125,   10,  134,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,   83,   16,  151,   18,  226,    9,   28,   65,  125,  170,  162,
            7,  125,   10,  228,   16,   30,   18,  162,   10,   17,  114,    9,
           10,  116,   16,   17,   18,   16,   20,   18,  170,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 136,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           27,   91,  125,  257,   23,   26,   27,  170,  257,   23,   93,  175,
          360,  104,   91,   13,   19,   17,   93,   15,   16,   17,   18,   34,
           79,   15,   16,   17,   18,   10,   21, 2117,  109,  525,   65,  457,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,  557,   63,  153,  154,   62,  153,  466,   68,  142,  154,  269,
         1088, 1182,  301,  239, 1485,   10,   13, 1486,  142,    9,   28,  362,
          358,   13,  112,  113,   91,   13,   93,   79,   10,  113,   91,   13,
           93,  797,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  113,   91,   13,   93, 1423,  573, 1222,   91,   23,   26,   34,
          887,   93,  251,  799,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  231,  129,  355,  125,  822,  130,  353,   91,   27,   93,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93, 1423,   91,   19,  887,   26,   34,  887,
           93, 1066,  113,   91,   29,   19,   13,   93,   10,  113,   91,   29,
           34,   13,   93,  113,   91,   29,   93,   10,   17,  113,   91,   29,
           93,   34,  113,   91,   17,   93,   34,  113,   91,   20,   93,   34,
          721,   34,  113,   91,  203,   93,   91,   27,   93,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    9,   28,  157,   62,  148,  362,  239,  958,   10,  201,   15,
           16,   82,   18,   65,  148,  265,  246,   62,  822,  151,  247,  168,
         1426,  157,   62,  158,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   49,   47,  142,  822,   63,  102,   55, 1007,   56,  591,   59,
           63,  153,   10,  154,   10,  116,   16,   29,   18,   16,   17,   18,
          594,  239,   83,   16, 1179,   18,   10,  239,   83,   16,  151,   18,
           10,  201,   15,   16,   82,   18,  241,  107,  154, 1599,  242, 1007,
          145, 1263,   53,  228,   16,  133,   18, 1600,   47,  142,  822,   63,
          247,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   11,   12,  160,  161,   16,
         1079,   18,   13,   34,   29,   26,  744,   13,  578,   23,  164,   17,
           15,   16,   13,   18,   26,  744,   13,  257,   23,  166,  161,   24,
          167,  113,   91,   13,   93,   34,  113,   11,   91,   13,   19,  116,
           16,   29,   18,   16,   17,   18,   24,   93,  257,   29,   13,  254,
          179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  32,  485,   16,   33,   18,   91,   29,   34,  296,   16,   35,   18,
           93,   10,   29,   19,  296,   16,   35,   18,   33,   10,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 738, 1439,  209,  894, 1440, 1441, 1442, 1443,  975, 1444,  409, 1441,
          530,  407,    9,   26,   28, 1369, 1370,  216,    9, 1418, 1441, 1445,
         1440, 1446,   28, 1418, 1441, 1447, 1440, 1446, 1441, 1448, 1446,  247,
         1449, 1450, 1451,  406, 1452,  413, 1453,  794, 1443, 1441, 1442, 1440,
         1454, 1444,   90, 1103,  301,   65, 1455,  209, 1440, 1443, 1442, 1454,
         1301, 1456, 1103,  142,   91,   65,   93,  430, 1457, 1458, 1440, 1443,
         1454, 1103,  102,  125,   26,  170,  610,  219,  212,  209, 1440, 1443,
         1454, 1459,  486,  102,   90,  545,  546,  547,  548,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   11,   12,  160,  161,   16,
         1079,   18,   17,   15,   16,   19,   13,   18,   26,  744,   13,  578,
           23,  164,   29,   26,  744,   13,  257,   23,  166,  161,   24,  167,
          113,   91,   13,   34,   29,   93,   22,  113,   91,   17,   13,   93,
           13,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   11,   12,   91,   13,   26,   79,   93,   14,
           13,   26,   79,  294,  465,   15,   16,  506,   18,   26,   79,  589,
           13,   24,   25,   28,   10,   12,   91,   13,   26,   79,   93,   14,
           13,   34,   79,   10,  134,   25,    9,   31,   28,  102,  484,  307,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  288,  645,  646,  647,  230,   11,   91,  288,   34,  116,   16,
          117,   18,   16,   21,   18,   24,   93,   10,  116,   16,   20,   18,
           16,   30,   18,  521,   11,   91,  288,   19,  116,   16,  117,   18,
           16,   21,   18,   24,   93,   10,    4,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 61,  13,  79, 159,  11,  12, 160, 161,  16, 277,  18,  17,  13,  19,
          79,  34,  29, 589,  23, 164,  13,  19,  17,  79,  19,  29, 578,  23,
         164,  13, 578,  29, 166, 161,  24, 167,  33,  10,  17,  13,  34,  20,
          79,  19,  30, 111, 179,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [891, 892, 893, 190,   9, 190,  28, 212, 209, 894, 895, 893, 221, 190,
           9, 212, 896, 897,  83,  16, 898,  18, 209, 896,  29,  83,  16, 898,
          18,  30, 899, 893, 221, 190,  28, 212, 896, 900,  83,  16, 898,  18,
         209, 896, 901,  83,  16, 898,  18,  20, 899, 893, 221, 190,   9, 832,
         902, 202, 893, 221, 190,  28, 832, 903, 202, 892, 437, 212, 896, 227,
          83,  16, 898,  18, 209, 896, 201,  83,  16, 898,  18, 339, 904, 899,
          86, 893, 190,   9, 190,  28, 832, 168, 905, 202,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 61, 180,  12, 125,  92, 131,  25, 125,  92,  16,  29,  18,  34,  20,
         125,  92,  16,  17,  18,  34, 721,  34,  91,  17, 131,  19,  29,  93,
         125,  92,  16, 131,  18,  10,  17, 131, 229,  12, 125,  92, 131,  25,
         526, 527, 229, 180,  12, 116,  16, 125,  92, 131,  18,  16,  17, 131,
          34,  29,  18,  25, 181, 131, 183,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 61, 136,  65, 137, 116,  16,  13,  15,  16,  17,  18,  18,  16, 125,
          15,  16,  17,  18,  18,  19, 116,  16,  79,  15,  16,  17,  18,  18,
          16, 170,  15,  16,  17,  18,  18,  10,  29,  91, 125, 257,  23,  26,
         170, 257,  23,  93, 258, 259, 139, 135,  92,  16,  29,  18,  26, 135,
          92,  16,  17,  18, 457, 228,  16,  30,  18, 142,  65, 238, 135,  92,
          16,  29,  18, 142, 172, 135,  92,  16,  17,  18, 142, 145, 146,  92,
          16,  29,  18, 135,  92,  16,  17,  18,  53, 125,  10,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,   99,  882,  102,    9,  584,   28,  882,  138,  233,  464,  584,
          465,  584,  568,  558,  466,   99,  882,  102,  107,  153,  242,  992,
          382,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 384,  645,  646,  647,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  173,   79,   10, 1112,   15,   16,   13,   18,   34,   13,  174,
           13,   68, 1112,  579,  176,  178,   79,   10,   17,   13,   34,  170,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 513,   65,  233,  234, 1775,   65,    9,   28,  362,  361,   65,  458,
           66,  256,  362,    7,   36,  153,   36,   10,   21,  228,   16,   17,
           18,   26,   27,   36,  560,   36,   10,   17,  228,   16,   30,   18,
           65,  139,  458,  358,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   91,   13,   19,   17,   93,
          248,   15,   16,   13,   18,   34,  125,   91,   13,   19,   29,   93,
           15,   16,   17,   18,  839,  250,  113,   91,   13,   93,  251,  252,
          842,  113,   91,   13,   93,  138,  253,  229,  125,  254,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  513,   65,   79,   15,   16,   17,   18,   10,   20,   13,  139,
          135,  177,  116,   16,   20,   18,   16,   17,   18,  107,  277,   65,
          267,    9,   28,   13,  456,  267,  142,   36, 1162,   36,   34,   36,
          562,   36,   10,   21,  229,  277,  178,  261,   16, 1963,   18,   10,
           20,  261,   16,  268,   18,  229,   36,  153,   36,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  107,   79,   10,   13,   34,   17,  125,  104,   65,  137,   13,
           15,   16,   17,   18,   34,   79,   15,   16,   17,   18,   19,   17,
          890,   19,   17,   10,   23,  374,    9,   26,   28,  362,   36,  153,
           36,   10,   17,  228,   16,   20,   18,  104,   65,   53,    4,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 123,  694,  695,  696,  102,  694,  125,   92,   16,   29,   18,  125,
           92,   16,   17,   18,   94,  125,   92,   16,  131,   18,   94,  125,
           92,   16,   29,   18,  294,   12,   23,   26,   29,   25,   91,   35,
           10,   29,   26,   17,   26,   94,   93,  697,  171,  125,   92,   16,
           35,   34,  171,   18,   10,  125,   92,   16,   35,   18,   91,   35,
           10,   29,   26,   17,   26,   94,   93,  698,  303,  123,  180,  699,
          125,   92,   16,   35,   34,  171,   18,   10,  125,   92,   16,   35,
           18,   91,   35,   10,   29,   26,   17,   26,   94,   93,  122,  697,
          171,  694,  123,  123,  171,  694,  125,   92,   16,   29,   18,  125,
           92,   16,   17,   18,   94,  125,   92,   16,  131,   18,   94,   26,
           65,   91,  278,   93,   10,  116,   16,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 274,  275,  276,  102,  104,   65,  178,   91,   13,   34,  105,   93,
           15,   16,   17,   18,   34,   79,   15,   16,   17,   18,   10,  214,
          140,  141,  280,   13,  281,  282,  283,  284,  285,  272,  229,   65,
          286,  178,  107,  277,  273,  178,   11,   12,  160,  161,   16,  277,
           18,   13,   10,  279,  114,  384,  164,   79,   10,  279,  230,  384,
          166,  161,   24,  167,  279,  273,  277,   65, 1573,    9,   28,  362,
           36,  153,   36,   10,  228,   16,   96,   18,  229,  277,  177,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 136,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           91,  125,  257,   23,   26,  170,  257,   23,   93,  175,  360, 1160,
         1161,   15,   16,   82,   18,   65,  457,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301,   83,   16,  353,   18,   92,   16,  131,   18,  370,   11,   12,
          125,   92,   16,  131,   18,   24,   25,  181,  131,  183,   20,   83,
           16,  353,   18,   92,   16,   20,   18,   10,   83,   16,  353,   18,
           92,   16,   17,   18,   34,   83,   16,  353,   18,   92,   16,   21,
           18,  125,   92,   16,   29,   18,   10,   17,  125,   92,   16,   30,
           18,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  261,   16,  153,   18,   10,   91,   17,   26,   20,   93,  261,
           16,  155,   18,   10,   91,   20,   26,  279,   93,  261,   16,  153,
           18,  263,  261,   16,  155,   18,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 764,   40,  766,  767, 1478, 2094, 2095, 2096, 2097, 2098, 2099, 2100,
         2101, 2102, 2103, 2104, 2105, 2106, 2107, 2108, 2109, 2110, 2111, 2112,
         2113, 1397, 2114,  102,  674, 2115,  674, 2115, 2043, 2098, 2116,  682,
         2109,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1177, 1178,   63,   19,    9,   92,   16,   29,   18,   28,   92,
           16,   29,   18,   65,   92,   16,   29,   18,   66,   92,   16,   29,
           18,   59,  812,  468,   92,   16,   29,   18,   10,   21,   26,   27,
          153,   10,   17,   26,   27,  239, 1179,   10,  351,   15,   16,   82,
           18,   26,   27,  256,   26,   27,  464,   26,   27,  465,  154,   26,
           27,  462,   92,   16,   29,   18,   26,   27,    9,   92,   16,   29,
           18,   66,  466,  241,  470,  469,  469,  242,   65,   92,   16,   29,
           18,  560, 1180,   65,  242,   65,   92,   16,  502,   18,  560,  358,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 107,  277,  173,   79,   10,  228,   16,   13,   18,  104,   13,   15,
           16,   17,   18,   34,   79,   15,   16,   17,   18,   10,  116,   16,
           29,   18,   16,   30,   18,  357,  277,  178,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   33,   10,  116,   16,   29,   19,   35,   18,   16,   29,   34,
           35,   18,   34,   17,   35,   36,   33,   36,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  353,   92,   16,  131,   18,  370,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,  125,   92,   16,   29,   18,
           10,   19,   17,   26,   27,  125,   92,   16,   17,   18,   34,  125,
           92,   16,  105,   18,   10,   17,   26,  353,   92,   16,   96,   18,
           10,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   79,  159,  160, 1740,   13,   19,   29,  589,   23, 1741,   19,
           79,  578,   23, 1741,   34,   79,   19,   21,  578,   23,  166, 1740,
          116,   16,   79,   18,   16,   13,   18,  168,  905,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    9,  584,   28,  255,  162,  137,  116,   16,   13,   15,   16,
           17,   18,   18,   16,   20,   18,   34,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  171,   18,   10,   29,  998,  138,  690,   65,
           68,  464,  239,   83,   16,  999,   18,   10,  819,   15,   16,   82,
           18,  171,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  104,   65,  178,   91,   13,   34,  105,   93,
           15,   16,   17,   18,   34,   79,   15,   16,   17,   18,   10,  214,
          140,  141,  280,   13,  281,  282,  283,  284,  285,  272,  229,   65,
          286,  178,  107,  277,  273,  178,   11,   12,  160,  161,   16,  277,
           18,   13,   10,  885,   16,  279, 1513,   18,  384,  164,   79,   10,
          885,   16,  279,  886,   18,  384,  166,  161,   91,  279,  478,   16,
          479,   18,   93,   24,  167,  277,   65, 1573,    9,   28,  362,   36,
          153,   36,  228,   16,   96,   18,  229,  277,  177,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   11,   91,   17,   34,   13,
           34,  641,   15,   16,   17,   18,   24,   93,  174,   91,   29,   34,
           13,   93,   19,   17,   13,  125,   10,   23,   26,  241,   19,   29,
           22,   13,   22,   23,  249,   26,  113,   91,   13,   93,   22,   23,
         1808,   13,  257,   23,  249,  113,   91,   13,   93,  257,   23,   13,
          113,   13, 1964,   68,  229,  125,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   32,   33,   91,   29,   34,   35,   93,   27,   33,   10,   17,
           35,   36,   33,   36,   10,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 242,  811,   63,  102,  239,    9,   10,  239,   28,   10,  239,   65,
           10, 1742,   15,   16,   82,   18,  154,   10,   17,  153,  254, 1739,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   13,   36,   13,   36,  822,   36,   13,  822,  125,
           36,  125,  125,  249,  229,  290,  113,   13,  293,  113,   13,  797,
           13,  456,  858,   56,   53,  754,  229,  125,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,  151,  102,  114,   65,   10,  116,   16,   17,   18,   16,   20,
           18,   26,  155,   10,   21,   26,  154,   10,   20,  114,   28,   10,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  11,   91,   13,   15,   16,   17,   18,   34,  116,   16,   17,   18,
           16,   13,   18,   24,   93,   15,   16,   30,   18,  582,  102,  296,
           16,   13,   18,   15,   16,   21,   18,  583,    4,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  135,  136,   65,  137,  116,   16,   13,   15,   16,   17,   18,
           18,   16,   21,   18,   19,  116,   16,   79,   15,   16,   17,   18,
           18,   16,   30,   18,   10,   29,  139,   68,  142,   65,   62,  140,
          141,   36,  144,   36,   10,   36, 1416,   36,  145, 1417,   53,    4,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   10,   12,   29,   26,   17,   26,   21,   25,   26,
           28,   10,   11,   12,   13,   36,   13,   15,   16,   17,   18,   19,
           21,   13,   34,  171,   10,   23,   24,   25,    9,   31,   28,   10,
           12,   29,   25,   28,   10,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  7, 169, 889,  16, 125,  18,  26, 889,  16, 170,  18,  36, 889,  16,
         125,  18,  36,  10,  29, 889,  16, 125,  18, 263, 889,  16, 170,  18,
          10,  19,  29, 889,  16, 125,  18, 263,  91,  17, 889,  16, 125,  18,
          19, 889,  16, 170,  18,  93,  10,  91,  27,  93,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [824, 825, 826, 406, 190, 827, 828, 829, 827,  13,  81, 830, 829, 831,
          79,  81, 279, 829, 832,  33,  81, 830, 833, 834, 829, 829, 827,  13,
          92,  35, 829, 831,  79,  92,  35, 990,  35,  10,  29, 584,  17, 584,
         837, 584, 134, 991,  90, 585, 508,  95, 217, 322, 610,  95, 554,  79,
          10, 125,  34, 752,  79,  10, 162,  34, 541, 228,  16,  13,  18, 103,
         829, 831,  79, 829, 827,  13, 100, 101, 446, 554, 838, 449, 839, 554,
         840,  90, 284,  79,  13, 100,   7, 190, 829, 832,  33,  13,  79,  84,
          33,  10, 841,  79,  19,  13, 842, 843,  35, 829, 827,  13,  10, 844,
         249, 829, 831, 829, 832, 845, 610, 846, 829, 827,  13, 847, 829, 832,
         845,   0],
        [107, 277, 255, 233, 139, 255,  64, 277, 358, 875, 116,  16,  29,  18,
          16,  21,  18, 255, 457,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [107,  79,  13, 104,  13,  15,  16,  17,  18,  34,  79,  15,  16,  17,
          18,  34,  17,  79,  19,  20,  10,  23, 361,   9,  26,  27,  28, 362,
          36, 153,  36,  10, 179,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,   83,   16,  151,   18,  226,    9,  584,   28,  584,   65,  125,
          584,  170,  584,  162,  584,    7,  230,   28,   34,  230,    9,   91,
          230,   65,   19,  114,   65,   93,   10,   23,  125,   10,   17,  162,
           10,  228,   16,   17,   18,   65,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 528, 1361, 1427, 1428, 1429,   20,   66, 1430,  696, 1431,  489,  231,
          489,  750,   63,   19,    9,   92,   16,   29,   18,  154,   92,   16,
           29,   18,   66,   92,   16,   29,   18, 1432,   49,   47,   62,   19,
         1433,  877,  519,   62,  750,   64,  256,   26,  135,   26,  260,   26,
          952,  558,  466,  153,   10,  154,   10,  105,   83,   16, 1019,   18,
           26,  468,   92,   16,   29,   18,   10,   21,   83,   16, 1019,   18,
           20,   66, 1430, 1434,  205, 1435, 1436,  477,  469, 1019,   15,   16,
           20,   18, 1430, 1437, 1431,  489,  828,  205, 1438,    4,  477,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  214,   18,   34,  116,   16,   79,   15,   16,   17,   18,   18,
           16,  171,   15,   16,   17,   18,   18,   10,   29,   91,   23,   22,
          171,   22,   30,   93,  457,  116,   16,  228,   16,  483,   18,   18,
           16,   21,   18,   26,    9,   26,   28,   65,  233,  229,   65,  178,
          567,  142,   65,   68,  568,  107,   13,   10,  105,   36,  569,   36,
           10,   36,  570,   36,   26,  569,  172,  570,  229,  145,  571,   53,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  169,  125,   10,   91,   29,   26,   19,   29,   93,   26,   27,
          170,   10,   91,  171,   34,   29,   26,   17,  171,   19,   21,   93,
          125,  172,  170,  171,   10,    4,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  33,   10,   29,   34,   17,  116,   16,   21,   35,   18,   16,   33,
           15,   16, 1999,   18,  425,   16,   33,   18,   19,   29,   18,   10,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  885,   16,  125,  114,   18,
           17,   13,   34,   91,  125,   19,   29,   93,   91,  114,   13,   34,
           29,   93,  125,  301,   36,  113,   91,   13,   93,   36,  168,    9,
          229,  113,   15,   16,  522,   18,   91,   13,   93,  229,    9,  241,
           36,  113,   15,   16,  522,   18,   91,   13,   93,   36,  578,   17,
            9,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,   83,   16,  151,   18,  102,  594,  154, 1067, 1068,  256,  594,
          466,  261,   16,   83,   16,  472,   18,   18,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  125,   10,  126,   92,   16,  841,   18,  901,   26,   27,  170,
           10,  126,   92,   16,   17,   18,  901,   26,   27,    4,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  136,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  125,   15,   16,   17,   18,   18,   19,  116,   16,   79,   15,
           16,   17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,
           29,   91,  125,  257,   23,   26,  170,  257,   23,   93,  175,  360,
           79,   10,  228,   16,   17,   18,   13,   65,  457,    4,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,  643,   10,   12,   29,   26,   17,   26,   20,   26,   21,
           26,   30,   26,  105,   26,  133,   25,   26,   27,    9,   10,   12,
           17,   26,   20,   26,   21,   26,   30,   25,   26,   27,   28,   10,
           12,   17,   26,   20,   26,  105,   26,  133,   25,   28,   31,   83,
           16,   65,   18,   92,   16,  643,   18,    9,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 309,  310,  311,  312,  313,  314,  315,  316,  317,  318,  319,  320,
          321,  320,  322,  323,  324,  325,  320,  326,  327,  328,  329,  330,
          321,  331,    7,  314,  332,  326,  333,  319,  320,  334,  335,  336,
          337,  338,  320,  339,  340,  319,   71,  341,  328,  342,  203,  319,
          320,  321,  337,  343,  320,  344,  320,  321,  345,  346,  347,  348,
          328,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1124, 1884, 1885, 1886,   76, 1887, 1888,  149, 1889,  149, 1889,
          604,  395,  151, 1890,  154, 1880,  153,  155,  145,   83,   16,  151,
           18, 1891,  858, 1419, 1892,  502,  983,  301,  842, 1893,  301,  979,
          102,  530,  987,  238,  988,  989,  839,  842,   71,  301,  373,   92,
           16,   29,   18,  373,   92,   16,   17,   18,  373,   92,   16,   20,
           18,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[127, 798, 129, 355, 279, 436, 130, 131,   3,   3,   3,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 13,  79, 159,  11,  12, 160, 161,  16, 277,  18,  13,  19,  79,  34,
          29, 165,  23, 164,  13,  19,  17,  79, 163,  23, 164,  13,  34,  17,
          79,  19,  17, 163,  23, 166, 161,  24, 167,  33,  13,  79, 168, 179,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 13,  26,  79, 159,  11,  12, 160, 161,  16, 277,  18,  13,  34,  79,
         165,  23, 164,  17,  13,  19,  79, 165,  23, 164,  13, 163,  29, 166,
         161,  24, 167,  33,  10,  20,  13,  34,  21,  79, 168,   4,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [127, 129, 355,  13,  79, 131, 130,  13,  79, 610,   3,   3,   3,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,  248,   15,   16,   13,   18,
           19,  125,   91,   13,   34,   17,   93,  125,   10,   29,  249,  250,
          113,   91,   13,   93,  251,  252,  113,   91,   13,   93,  138,  253,
          229,  125,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   12,   13,   36,   13,   19,   29,  589,   23,
           25,   26,   27,   28,   10,   12,   23,   26,   27,   29,   26,   27,
           17,   25,   26,   27,    9,   31,   28,   10,   91,   27,   93,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,  230,   13,   19,  228,   16,   20,   18,  114,   13,
          797,  112,   79,   10,  230,   13,   34,  228,   16,   20,   18,  114,
           13,  797,  348,  259,  876,    4,   81,  110,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  370,  164,   16,  181,  183, 1763,  125, 1764,  125,   92,   16,
          216,   18,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  521,   11,   91,  384,   19,  116,   16,   30,  117,   18,   16,
           21,   18,   24,   93,   10,  116,   16,   29,   18,   16,   30,   18,
           26,  521,  384,   10,    4,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93, 1222,   13,  859,   23,  249,  113,   91,
           13,   93,   10,  248,   15,   16,   19,   13,   19,   29,   18,   19,
           13,  173,   79,   10,  113,   91,   13,   93,   68,  579,  176,  178,
          179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 750,   83,   16,   63,   18,   19,   83,   16,    9,   18,   92,   16,
           29,   18,   83,   16,   28,   18,   92,   16,   29,   18,   83,   16,
           65,   18,   92,   16,   29,   18,   83,   16,   66,   18,   92,   16,
           29,   18,  102,  153,   10,  154,   10,   17,  155,   92,   16,   29,
           18,  242,   83,   16,  462,   18,   92,   16,   29,   18,   83,   16,
           65,   18,   92,   16,   29,   18,   83,   16,   65,   18, 1074, 1075,
          596,   15,   16,   82,   18,  750,  247,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   65,   92,   16,   29,   18,   26,   65,   92,   16,   17,   18,
          273,  178,   65,   92,   16,   29,   18,  137,   11,   12,  160,  161,
           16,  277,   18,   13,   10,   21,  114,   15,   16,   17,   18,  288,
          164,   79,   10,   21,  230,   15,   16,   17,   18,  288,  166,  161,
           26,   91,  288,  478,   16,  479,   18,   93,   26,   65,   92,   16,
           17,   18,  137,   11,   12,  160,  161,   16,  277,   18,   13,   10,
          279,   34,  116,   16,   29,   18,   16,  279,   18,  164,   79,   10,
          279,   19,  116,   16,   29,   18,   16,  279,   18,  166,  161,   26,
           91,  279,  478,   16,  479,   18,   93,   24,  167,   24,  167,   65,
           92,   16,   29,   18,   26,   27,   65,   92,   16,   17,   18,  273,
          178,  480,  178,  140,  141,  280,   13,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  513,   65,   79,   15,   16,   17,   18,   10,   17,   13,   26,
          106,  107,  277,  361,   65,    9,   28,  362,  104,  464, 1089,  153,
          604,  104,  241,  140,  141,   62,  104,  464, 1801,  464,  106,  142,
          229,  107,  277,  104,  464,  178,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1213, 1214,   52,   51, 1215, 1216,  292,  688,  520,  688,  158,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 976,  977,  978,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301, 1424,  370,   11,   12,  125,   92,   16,  131,   18,   24,   25,
          181,  131,  183,  125,   92,   16,   20,   18,   10,   30,   26,  125,
           92,   16,  133,   18,   10,  729,  353,   92,   16,   96,   18,   10,
         1425,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    9,  584,   28,  584,   65,  147,   62,  148,  149,   68,   27,
          150,   62,   92,   16,   29,   18,  363,  151,  152,  150,   62,   92,
           16,   29,   18,   53,   21,  117,  153,   10,  154,   10,  155,   10,
          156,   92,   16,   29,   18,   27,  157,   62,  158,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    9,   28,  136,  256,  258,  259,  233,   68,  464,  256,  145,
           83,   16, 1110,   18,  957,  669, 2038,  819,  820,  256,  457,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   66,  145,  151,  242,  238, 1735,   16, 1736,   18,   16,  154,
           18,   10,   20, 1735,   16, 1736,   18,   16,  269,   18,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 370,   12,   11,  167,  125,   92,   16,  131,   18,   24,   25, 2008,
           29,   26,   27,  795,  125,   92,   16,   17,   18,   26,   27,  125,
           92,   16,   20,   18,   26,   27,  125,   92,   16,  105,   18,  985,
          132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  181,  183,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 980, 1682, 1683, 1684, 1685, 1683, 1684,  548, 1685,   44, 1686,  750,
          882, 1687, 1688, 1125, 1689, 1690, 1691, 1685,   44, 1383, 1692,  121,
         1383, 1692,  894,  894, 1693,  858,  688, 1383, 1692, 1694,  766,  736,
         1695,  121, 1696, 1697, 1383, 1692,  233, 1284,  882, 1069,  882, 1698,
         1383, 1692,  682, 1699, 1700, 1701, 2185,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,   19,   13,   15,   16,   21,   18,   34,   13,   15,
           16,   17,   18,   34,   17,  797,    4,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   11,   91,   29,   19,   13,
           15,   16,   17,   18,   24,   93,  248,   15,   16,   13,   18,  250,
          113,   91,   13,   93,  251,  252,   13,  589,   23,  249,  113,   91,
           13,   93,  578,  641,   34,   29,  229,  125,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 125,  861,  116,   16,   17,   34, 1471,   18,   16,   29,   34,   35,
           18,   10,   20,   34,   35,   26,   27,  125,   10,   91,   27,   93,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 891,  892,  893,  190,    9,  190,   28,  212,  209,  894,  895,  893,
          221,  190,    9,  212,  896,  897,  898,  209,  896, 1786,  899,  893,
          221,  190,   28,  212,  896,  900,  898,  209,  896,  901,  898,  899,
          893,  221,  190,    9,  832,  902,  202,  893,  221,  190,   28,  832,
          903,  202,  892,  437,  212,  896, 1787,  209,  896, 1788,  339,  904,
          899,   86,  893,  190,    9,  190,   28,  832,  168,  179,  202,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   30,   19,   36,   13,   34,
          125,   36,   19,   36,   13,   19,   17,   36,  125,   10,   29,  249,
          229,  290,  113,   91,   13,   93,  589,   23,  293,  113,   91,   13,
           93,  578,   29,  229,  125,  254,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 242,  274,  275,  276,  102,  150,   62,  273,  178,   11,   12,  160,
          161,   16,  277,   18,   83,   16,   13,   18,   10,  114,  288,  164,
           79,   10,  230,  288,  166,  161,   26,   91,  288,  478,   16,  479,
           18,   93,   26,  478,   16,  106,   18,   91,   23,   26,   24,  167,
           19,  228,   17, 1160,  384,  107,  277,  150,   62,  361,    9,   28,
          362,  229, 1393,  254,  229,  153,  466,  142,  512,  273,  178,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  128,  129,  355,  278,   10,   23,   26,   27,  125,   10,   23,
           26,  130,  278,    3,    3,    3,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[2009, 1285,  255,   65,  116,   16,   13,   15,   16,   17,   18,   18,
           16,  125,   15,   16,   17,   18,   18,   34,  116,   16,   79,   15,
           16,   17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,
           29,   27,   91,  125,  257,  170,  257,   23,   93,  258,  259,  233,
            9,   92,   16,   29,   18,   26,   27,    9,   92,   16,   17,   18,
           26,   27, 1089,    9,   92,   16,   29,   18,   83,   16,    9,   18,
           92,   16,   17,   18,  604,  104,  107,   83,   16,  752,   18,   19,
           83,   16,  890,   18,   34,   17,   83,   16,  575,   18,   10,   23,
          357,   83,   16,   65,   18,  457,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  226,    9,   26,   27,   28,   26,   27,   65,  125,   26,
          170,   26,  162,    7,   28,   10,  227,   15,   16,   82,   18,  125,
           10,  228,   16,   20,   18,  162,   26,   27,  170,   10,   17,  228,
           16,  133,   18,   26,   27,  229,  145,  151,   53,  230,    9,   34,
          228,   16,   20,   18,  230,   65,   10,  116,   16,  228,   16,   17,
           18,   18,   16,   17,   18,   26,   27,  229,   65,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  91,   13,   15,   17,   34,   13,   34,   79,   93,   15,   30,  582,
          102,   13,   15,   30,   79,   15,   17,  583,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1188, 1189, 1190, 1191, 1192, 1059, 1193, 1050,  102, 1194,  544, 1192,
          544, 1195, 1196,  121,  593,    9,   68, 1197,  544, 1192,  510,  483,
           28,   68, 1198,  544, 1195,  510, 1199, 1063,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  169,  261,   16, 1187,   18,   10,   11,   91,  116,   16,   29,
           18,   16,   17,   18,   26,  116,   16,  228,   16,   20,   18,   18,
           16,   17,   18,   24,   93,  261,   16,   83,   16,  154,   18,   18,
           10,   11,   91,  116,   16,  228,   16,   20,   18,   18,   16,   17,
           18,   26,  116,   16,   29,   18,   16,   17,   18,   24,   93,  239,
           83,   16,  151,   18,   10,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 116,   16,   20,   34,   35,   18,   16,   29,   34,   35,   18,   10,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 173,   79,   10,  174,   13,   34,   13,   34,   29,  175,  176,  177,
          176,  178,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,   19,
           17,   79,   19,   17,  578,   23,  164,   13,   19,   79,   34,   29,
          589,   23,  164,   79,  163,   23,  166,  161,   24,  167,   33,  807,
         1840,  168,  179,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    9,  255,  256,  137,  116,   16,   13,   15,   16,   17,   18,
           18,   16,   21,   18,   34,  116,   16,   79,   15,   16,   17,   18,
           18,   16,   20,   18,   10,   29,  258,  233,  177,  278,  278,  107,
          361,  256,    9,  464,  362,   68,  465,  256,   83,   16, 1183,   18,
          172,   83,   16, 1567,   18,   83,   16,  467,   18,   36,   10,   36,
           83,   16, 1568,   18,  249,  229,  363,  471,   53,   36,  467,   36,
           36, 1568,   36,  249,  241,  228,   16,   20,   18,   22,   83,   16,
          278,   18,   22,   17,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 173,   79,   10,   91,  641,   34,   29,   93,  248,   15,   16,   13,
           18,   68,  579,  176,  177,  125,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 314,  917, 1194, 1286,  447, 1287, 1288, 1289, 1290, 1291,  202, 1292,
         1291,  202, 1094, 1293, 1287, 1294, 1291,  202, 1295,  534, 1296, 1297,
         1298, 1299, 1300,  534, 1192,   81, 1192, 1301,  214, 1300,  333, 1302,
         1192, 1274,  546,   85,  214, 1300, 1303, 1302, 1192, 1301,   85, 1300,
          409, 1302, 1304, 1286,  917,  217,  181,  922, 1304, 1305, 1192,   90,
         1100, 1101, 1102, 1192, 1274,  546,  942,  219, 1192, 1274,  546,   71,
          229, 1304, 1287, 1300,  339, 1303, 1302,   71,   61, 1304, 1298, 1287,
          832, 1306,   81,  202, 1304, 1287, 1288, 1307, 1308, 1269, 1306,  610,
          219, 1306,  754, 1309,   71,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  750,   63,   19,    9,   92,   16,   29,   18,   28,   26,   65,
           92,   16,   29,   18,   66,  102,   68,  256,  135,  558,  559,   92,
           16,   29,   18,   26,   27,  462,   92,   16,   29,   18,   17,  560,
           10,  561,   92,   16,   29,   18,  562,   10,   17,  563,  241,   68,
           65,   92,   16,   29,   18,  242,  565,  751,  153,   10,   17,   26,
           27,  594,   10,   29,   26,   27,  468,   92,   16,   29,   18,   10,
           20,  229,  607,    9,   19,  564,   19,    9,   92,   16,   29,   18,
          720,  610,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   17,  230,   13,   19,   13,
          114,   13,   19,   13,   26,  113,   15,   16,  522,   18,   91,   13,
           93,  113,   91,   13,   93,  369,  241,  113,   15,   16,  522,   18,
           91,   13,   93,  546,   91,   23,   26,  117,   93, 1181,  253,   13,
          294,  118,   23,   26,  117,  119,  249,  113,   91,   13,   93,  165,
          641,  229,  125,  254,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61, 1393, 1662,  138,  242, 1393, 1662, 1515,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 488,  489,  490,  491,  492,  493,  494,  495,   90,  284,  397,  309,
          310,  496,  497,  498,  433,  499,   91,  279,   93,  279,   81,  500,
          488,  489,  501,  502,   91,  279,   93,   10,  116,   16,  503,   18,
           16,   29,   34,  248,   15,   16,   19,  504,   91,  279,   19,  505,
           93,   18,   18,  503,  497,  498,  433,  502,   11,   91,  279,   15,
           16,  506,   18,   24,   93,   10,  347,  503,  507,  508,  509,  311,
          279,   15,   16,  506,   18,  510,  368,  174,  511,  452,   20,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   36,   17,   13,   34,   29,
           36,   34,   36,   13,   19,   29,   36,  292,   79,  113,   13,  797,
           83,   16,   13,   18,  294,  118,   23,   26,   34,  887,   93,  249,
          113,   91,   83,   16,   13,   18,   93,  578,   83,   16,  641,   18,
           34,   83,   16,  170,   18,   26,   27,  229,   83,   16,  125,   18,
           34,   83,   16,  170,   18,  111,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 33,  10,  21,  34,  20,  35, 116,  16, 425,  16,  33,  18,  18,  16,
          36,  33,  36,  18,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [231, 811,  63, 812, 260, 155, 813, 267, 814, 815, 242,  63, 241, 242,
         816, 815, 242, 817, 818, 151, 819, 820, 603, 815, 821, 246, 256, 822,
         823, 247, 116,  16, 228,  16, 105,  18,  18,  16,  20,  18, 429, 246,
         245,  53,   3,   3,   3,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [173,  79,  10,  20,  11,  91,  13,  15,  16,  17,  18,  34,  13,  24,
          93, 248,  15,  16,  13,  18,  68, 179, 579, 176, 178,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [697, 395, 375,  51, 303, 307, 486, 761, 762,  67, 433, 307, 486, 761,
         762,  71,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   11,   36,   13,   19,  116,
           16,   29,   18,   16,   17,   18,   24,   36,   34,   11,   36,   13,
           34,  116,   16,   29,   18,   16,   17,   18,   24,   36,  464,  290,
          113,   91,   13,   93, 1574,   17,  293,  229,  464,  241,  125,   26,
          170,  294,  464,  249,   36,  125,   34,  170,   36,   22,   36,   29,
           34,  575,   36,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  33,   91,   29,   34,   35,   93,   10,   17,   35,   33,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  140,  141,  280,   13,  456,  580,  282,  283,
          284,  285,  272,  173,   65,   92,   16,   29,   18,  286,  178,  373,
          114,  288,   10,   21,  464,  173,   83,   16,   65,   18,   92,   16,
           29,   18,  265,   68,  142, 1089,  848,   36,  848,   36,  263,   36,
          144,   36,   10,  182, 1180,  142,  512,   83,   16,   65,   18,   92,
           16,   17,   18,  274,  178, 1270,    9,  286,   11,   91,   17,   26,
          116,   16,  117,   18,   16,   20,   18,   24,   93,   68,   28,  173,
           83,   16,   65,   18,   92,   16,   17,   18,  229,  145,   83,   16,
          956,   18,   53,  168,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  116,   16,   29,   18,   16,   30,
           18,  230,   11,   91,   13,   34,  116,   16,  117,   18,   16,   20,
           18,   24,   93,   34,  114,   11,   91,   13,   19,  116,   16,  117,
           18,   16,  105,   18,   24,   93,  168,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   10,   12,   29,   26,   17,   26,   20,   26,   30,
           26,  133,   26,  482,   25,   26,   27,   28,   10,   12,   13,   14,
           20,   22,   13,   22,  483,   25,   26,    9,   31,   28,  102,  484,
          307,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  32,  116,   16,   29,   18,   16,   29,   19,   20,   35,   18,  722,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  248,   15,   16,   13,   18,
           19,  641,   15,   16,   17,   18,  125,   10,   29,  241,   13,  589,
           23,  249,  113,   91,   13,   93,  589,   29,  113,   91,   13,   93,
           91,   23,   26,   34,  887,   93,  253,  229,  125,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7, 2118,  460,   83,   16,  151,   18,   19,   83,   16,    9,   18,
           92,   16,   29,   18,   83,   16,   28,   18,   92,   16,   29,   18,
           83,   16,   65,   18,   92,   16,   29,   18,  102,  239,   83,   16,
          151,   18,   10,  819,   15,   16,   82,   18,  153,   10,   17,   83,
           16,  154,   18,   10,   83,   16, 1624,   18,   92,   16,   29,   18,
           10,   29, 1625,  107,   83,   16,  153,   18,   92,   16,   29,   18,
           83,   16,  154,   18,   92,   16,   29,   18,  719,  609,  741,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 271,  272,  273,  178,  231,  285,  272, 1469,  102,    9,   91,   17,
           26,   23,   93,   28,   11,   91,  228,   16,   17,   18,   26,  116,
           16,  117,   18,   16,   21,   18,   24,   93,   65,   11,   91,  228,
           16,   17,   18,   26,  116,   16,   20,  117,   18,   16,   21,   18,
           24,   93,   66,   91,   17,   26,  117,   93, 1470, 1245,   16,  153,
           18, 1245,   16,  154,   18, 1245,   16,  269,   18,  104,  234,   11,
           91,   29,   26,  116,   16,  117,   18,   16,   17,   18,   24,   93,
           91,   29,   26,  117,   93,  173,  464,   92,   16,   29,   18, 1470,
         1245,   16,  153,   18,  173,  464,   92,   16,   17,   18, 1470, 1245,
           16,  154,   18,  173,  464,   92,   16,   20,   18, 1470, 1245,   16,
          269,   18, 1269,  464,   92,   16,   29,    0],
        [ 285,  272,  102,   62,  280,   68,  464,   11,   91,  287,   92,   16,
           23,   18,   26,  288,   92,   16,   23,   18,   24,   93,   11,   91,
          287,   92,   16,   23,   18,  257,   23,   24,   93,  173,   65,  137,
          287,   10,   21,  230,  288,  107,  277,  106,    9,   91,   21,   26,
           23,   93,  848,  352, 1571,  142,  288,   92,   16,   23,   18,   10,
          116,   16,  117,   18,   16,   20,   18,  249,  229,  287,   92,   16,
           23,   18,  277,  286,  178,  464,   65, 1182,  142, 1089,  848, 1721,
          229,  142,   68,  512,  286,  178,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  135,   92,   16,   29,   18,   26,  135,   92,   16,   17,   18,
          255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   34,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           91,  125,  257,  170,  257,   23,   93,  138,  139,  142,   65,  238,
           62,  140,  141,  145, 1717,   92,   16,   17,   18,   56,  229,   65,
          457,   68,  142,  146,   92,   16,   29,   18,  172,  146,   92,   16,
           17,   18,  145,  135,   92,   16,   29,   18,  146,   92,   16,   17,
           18,   53,   54,  182,  229,  170,  610,  125,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 764, 1668, 1669,  696, 1670,  217, 1671, 1672, 1668, 1673,  102,   96,
         1674, 1675, 1676,   85, 1674, 1675, 1677,   96, 1674, 1675, 1678, 1671,
         1679, 1668, 1673, 1674,  544, 1675,  430, 1667,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,    9,   28,   65,   66,  256,  135,  233, 1791,  102,  718, 1792,
           52, 1162, 1793,  239,   83,   16, 1794,   18,   10,  201,   15,   16,
           82,   18,  607,   66,  822, 1162,  822,  256,  607,   65,  822,  814,
          822,  135,  351,   15,   16,   82,   18,  241,  242, 1792,  172,  242,
         1795,  229,  607,  256,  822,  154,  822,    9,  609,  610,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7, 1223, 1186,  180,   12,  125,   92,  131,   25,  125,   92,   16,
           29,   18,   10,   29,   26,  125,   92,   16,  131,   18,   15,   16,
           17,   18,   19,   11,   91,   17,  125,   92,   16,  131,   34,   29,
           18,   19,   29,   24,   93,  125,   92,   16,  131,   18,   19,   17,
          125,   92,   16,  131,   34,   29,   18,   10,   23,  229,  125,   92,
           17,   26,  125,   92,   20,  229,   12,  125,   92,  131,   25,  526,
          527,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,   83,   16,    9,   18,   10,   12,   29,   26,   17,   26,
           20,   25,   26,   83,   16,   28,   18,   10,   11,   12,   83,   16,
           13,   18,   36,   83,   16,   13,   18,   15,   16,   17,   18,   22,
          366,   24,   25,   26,   83,   16,    9,   18,   31,   83,   16,   28,
           18,   10,   91,   27,   93,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  248,   15,   16, 1312,   18,
           34,   13,   15,   16,   17,   18,   19, 1312,  241,  113,   91,   13,
           93,   91,   19,  887,   26,   23,   93,  251,  799,   91,   23,  584,
           34,  887,   93,  251,  888,  378,   13,   92,   16,   29,   18,   26,
           27,   13,   92,   16,   17,   18,  294,  118,   19,   29,   26,   27,
           29,  119,   36,  113,   11,   91,   13,   92,   16,   29,   18,   24,
           93,   19,  113,   11,   91,   13,   92,   16,   17,   18,   24,   93,
           36,  163,  248,   19,   29,  229,  171,  254,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,   13,  584,   79,  159,   11,   12,  160,  161,   16,  277,   18,
           13,   34,   20,   79,  163,   20,  164,   13,   19,   79,  165,   29,
          164,   79,  165,   23,  166,  161,   24,  167,   33,   10,   13,   34,
           79,  168,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  145,  151,   53,  116,   16,  366,  228,   16,   20,   18,   18,
           16,   21,   18,   56,  233,  157,   62,  148,  367,   62,  158,  182,
          117,   26,   62,  242,  151,  358,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  246,  142,   19,  151,  242,  592,  593,  155,   10,   29,   26,
           27,  153,   10,  594,   10,  228,   16,   20,   18,   26,   27,  153,
          172,  155,   26,   27,  153,  172,  594,   97,  114,  239,  595,   10,
          596,   26,  597,  114,  239,  598,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   10,   91,   29,   26,   17,   93,
           26,  889,   16,  170,   18,   10,   91,   17,   26,   19,   17,   93,
           26,  889,   16,  162,   18,   10,   91,   29,   26, 1348,   93,  261,
           16,   83,   16,  162,   18,   18,  469,  469,   91,   17,  261,   16,
           83,   16,  125,   18,   18,   34,  261,   16,   83,   16,  170,   18,
           18,   93, 1349, 1350,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  246,  142,  151,   99,  233,  157,   62,  148,  266,  268,  608,
          145,  151,   51,  236,  256,  135,  266,  268,  466,  239, 1941,   10,
          201,   15,   16,   82,   18,  157,   62,  247,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  150,  464,  137,   13,   15,   16,   17,   18,   34,   79,   15,
           16,   17,   18,   19,   17,   13,   19,   17,   79,   19,   17,   10,
           23,   26,  107,  277,  137,   17,   13,   34,   79,   34,   17,   10,
           23,   26,  142,  277,  265,  106,  142,  585,  150,  464,  176,  266,
          268,   26,  586,    9,   26,   28,   26,   36,  587,   36,  263,   36,
          153,   14,  588,  107,  153,  178,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 309,  310,  311,  312,  313,  314,  315,  316,  317,  318,  319,  320,
          321,  320,  322,  648,  324,  325,  320,  326,  327,  328,  329,  330,
          649,    7,  314,  332,  326,  333,  319,  320,  334,  335,  336,  337,
          338,  320,  340,  319,   71,  341,  328,  342,  203,  319,  320,  321,
          337,  326,  320,  344,  320,  321,   71,  346,  347,  348,  328,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   65,  726,   79,   10,  116,   16,   13,
           15,   17,   18,   16,   21,   18,  107,  277,  726,   79,   10,  566,
           34,  125,   91,  125,  962,   23,   93,  515,  464,  465,  362,  278,
           10,   23,  249, 1752,   65,   68,  464,  465,  579,  176,  178,   79,
         1314,   68,  142,  278, 1753,  249, 1754,  239, 1755,   10,  239, 1756,
          449,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 738,    7,  173,   65,  137,   79,   10,  116,   16,   13,   15,   16,
           17,   18,   18,   16,   17,   18,   26,   66,  107,   79,   10,   19,
          116,   16,   29,   18,   16,   17,   18,  265,   66,  585,   65,  359,
          176,  586,    9,   26,   28,  241,  107,  153,  270,  256,   11,   91,
           23,   26,  116,   16,   30,   18,   16,   17,   18,   24,   93,  234,
          104,  107,  153,  357,  586, 1089,  153,  466,  429,  104,  178,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  798,  129,  355,  279,  436,  130,  131,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  136,   65,  116,   16,   13,   15,   16,   17,   18,   18,   16,
           20,   18,   19,   79,   15,   16,   17,   18,   10,   29,   62,  140,
          141,  135,   65,  259,  139,  135,  107,   65,  359,  360,  267,  464,
          465,  145, 1894,  395,   36,   83,   16,  470,   18,   36,   10,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 824,  825,  826,  406,  190,  827,  828,  829,  827,   13,   81,  830,
          829,  831,   79,   81,  279,  829,  832,   33,   81,  830,  833,  834,
          829,  829,  827,  835,  829,  831,  836,   35,  837,   90,  585,  508,
           95,  217,  322,  610,   95,  554,   79,   10,  125,   34,  752,   79,
           10,  162,   34,  541,  228,   16,   13,   18,  103,  829,  831,   79,
          829,  827,   13,  100,  101,  446,  554,  838,  449,  839,  554,  840,
           90,  284,   79,   13,  100,    7,  190,  829,  832,   33,   13,   79,
           84,   33,  841,   79,  822,   13,  842,  843,   35,  829,  827,   13,
          844,  249,  829,  831,  829,  832,  845,  610,  846,  829,  827,   13,
          847,  829,  832,  845,  610,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  353,   92,   16,  131,   18,  370,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,  125,   92,   16,   29,   18,
          572,   23,   26,   27,  125,   92,   16,   17,   18,   10,   20,  125,
           92,   16,   29,   18,  116,   16,  353,   92,   16,   96,   18,   18,
           16,  353,   92,   16,   30,   18,   18,   10, 2034,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,   19,
           17,   79,   19,   17,  163,   23,  164,   13,   19,   79,   34,   29,
          589,   23,  164,   79,  163,   23,  166,  161,   24,  167,   33,   10,
           20,   13,   34,   17,   79,  168,  179,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,
           34,   17,   79,  163,   29,  164,   17,   13,   34,   79,  589,   19,
           29,  164,   13,   19,   79,  578,   23,  166,  161,   24,  167,   33,
          807,  822, 1840,  111,  737,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1358,   13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   17,
           13,   34,   79,   34,   20,  165,   23,  164,   13,   19,   17,   79,
           34,   21,  165,   23,  164,   13,   19,   17,  163,   23,  166,  161,
           24,  167,   33,   10,   13,   34,  116,   16,   29,   18,   16,   20,
           18,   79,  168, 1350,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  353,   92,   16,  131,   18,  370,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,    7,  353,   92,   16,  366,
           18,   10,   19,  125,   92,   16,   30,   18,  125,   92,   16,   20,
           18,   10,   21,  229,   11,   12,  125,   92,   16,  131,   18,   24,
           25,  526,  527,  125,   92,   16,   29,   18,  257,   23,  229,  353,
           92,   16,  131,   18,  165,  125,   92,   16,  131,   18,  131,  254,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   32,   33,   92,   16,   29,   18,   33,   92,   16,   17,   18,
           11,   36,   33,   92,   16,   29,   18,   24,   36,   10,   11,   36,
           33,   92,   16,   17,   18,   24,   36,   10,   17,   33,   92,   16,
           29,   18,   34,   33,   92,   16,   17,   18,   10,  228,   16,   20,
           18,   34,  296,   16,   35,   18,   11,   36,   33,   92,   16,   29,
           18,   19,   33,   92,   16,   17,   18,   24,   36,   10,    4,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   10,   11,   12,   13,   14,   13,   15,   16,   17,
           18,   19,   21,  578,   23,   24,   25,   26,   27,   28,   10,   12,
           13,   14,   17,   13,   34,  125,  578,   23,   25,   26,    9,   31,
           28,   10,   12,   13,   14,   19,   17,  163,   13,  578,   29,   25,
           26,  125,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 423,  353,   10,   29,   19,  116,   16,   29,   18,   16,   17,   18,
           34,  116,   16,   29,   18,   16,   20,   18,   19,  116,   16,   29,
           18,   16,   21,   18,   34,  721,   34,  116,   16,   29,   18,   16,
         1622,   18,   19,  116,   16,   29,   18,   16,  216,   18, 1623,  231,
          129,  995, 1159,  997,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1217, 1177,  459,  460,   83,   16,  151,   18,   19,   83,   16,    9,
           18,   92,   16,   29,   18,   83,   16,   28,   18,   92,   16,   29,
           18,   83,   16,   65,   18,   92,   16,   29,   18, 1218,  247, 1219,
          157,  153,  172,  154,  153,   10,  105,   83,   16,  154,   18,   10,
          134,  468,   92,   16,   29,   18,   10,   20, 1219,  168,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1206, 1394, 1206, 1395,  530, 1396, 1206, 1395, 1397,   71,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   13,  822,   36,   13,  125,   36,  822,   36,   13,
          822,   36,  125,  249,  229,  290,  113,   91,   13,   93,  589,   23,
          293,  113,   91,   13,   93,  578,   29,  229,  125,  254,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  255,   65,   92,   16,   29,   18,  116,   16,   13,   15,   16,
           17,   18,   18,   16,  125,   15,   16,   17,   18,   18,   34,  116,
           16,   79,   15,   16,   17,   18,   18,   16,  170,   15,   16,   17,
           18,   18,   10,   29,   91,  125,  257,  170,  257,   23,   93,  259,
          139,  135,  513,   65,   92,   16,   17,   18,  139,  455,   65,   92,
           16,   29,   18,   64,   65,   92,   16,   17,   18,  233,  455,  135,
           13,  456,  352,  107,  361,   65,   92,   16,   29,   18,    9,   28,
          362,  361,   65,   92,   16,   17,   18,   65,   66,  362,   36,  269,
           36,   10,  116,   16,   21,   18,   16,   20,   18,   36,  153,   36,
          229,   65,   92,   16,   29,   18,  457,   61,  464,   65,   92,   16,
           29,   18,   65,   92,   16,   17,   18,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[271, 272, 273, 178, 274, 275, 276, 102, 173,  65, 273, 178,  11,  12,
         160, 161,  16, 277,  18,  13,  10, 116,  16,  29,  19, 279,  15,  16,
          17,  18,  18,  16,  29,  34, 279,  15,  16,  17,  18,  18, 164,  79,
          10, 116,  16,  21, 279,  18,  16,  29,  34, 279,  15,  16,  17,  18,
          18, 166, 161,  24, 167, 279, 273, 140, 141,  62, 280,  13, 456, 580,
         282, 283, 284, 285, 272, 107, 277, 286, 178,  17, 287, 114, 288,  34,
         228,  16,  20,  18, 287, 230, 288,  34, 482,  10,  23, 229,  65, 277,
         274, 178, 229,  65,  68, 277, 358, 111,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [112, 113,  91,  13,  93,  10, 114,  91, 115,  13,  34, 796,  93, 797,
         798, 113,  91,  13,  93, 251, 799, 546,   3,   3,   3,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [  7,  17, 521, 288,  19, 521,  11,  91, 288,  34, 116,  16, 117,  18,
          16,  21,  18,  24,  93,  10, 133, 521, 288,  10,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [173,  79,  10,  20,  11,  91,  13,  15,  16,  17,  18,  34,  13,  24,
          93, 248,  15,  16,  13,  18,  68,  91,  23,  26,  23,  93, 579, 176,
         178,   4,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 242,  274,  275,  276,  102,  150,   62,  273,  178,   11,   12,  160,
          161,   16,  277,   18,   13,   10,  114,  288,  164,   79,   10,  230,
          288,  166,  161,   24,  167, 1392,  273,  106,   91,   23,   26,   19,
          228,   16,   17,   18,   93, 1160, 1393,  107,  277,  150,   62,  515,
            9,   28,  362,  229, 1393,  254,  229,  153,  466,  142,  512,  273,
          178,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   91,   13,   19,   29,   93,
          174,   13,   19,   13,   19,   29,  241,  113,   91,   13,   93, 1181,
         1113,   68,  113,   91,   13,   93,   10,   23,  380,  381,  138, 1718,
          138, 1718, 1719, 1720,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,    7,  459,  460,  151,   19,    9,   92,   16,   29,   18,   28,
           92,   16,   29,   18,   65,   92,   16,   29,   18,   59,  236,   55,
          462,   92,   16,   29,   18,   65,   92,   16,   29,   18,   65,  463,
          464,  465,  154,   28,   92,   16,   29,   18,   65,   92,   16,   29,
           18,  466,  142,  467,  238,   28,   92,   16,   29,   18,   65,   92,
           16,   29,   18,  142,  242,  361,  153,  256,  361,  155,  135,  241,
          468,   92,   16,   29,   18,  717,  470,  242,    9,   92,   16,   29,
           18,  471,  172,  242,  472,   92,   16,   29,   18,   65,   92,   16,
           29,   18,  135,   61,   62,    9,   92,   16,   29,   18,   28,   92,
           16,   29,   18,   65,   92,   16,   29,   18,   64,  473,  717,  718,
          472,   92,   16,   29,   18,   65,   92,    0],
        [1189, 1190, 1275, 1276,  441, 1188, 1167, 1277,  218, 1225,  829, 1194,
         1253,  829,  298, 1194,  313, 1194, 1278, 1279,   81, 1280,   90, 1193,
         1230, 1230, 1062, 1281,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 396,  397,  398,  399,  400,  714,  402,  403,  404,  405,  397,  406,
          402,  403,  407,   53,  408,  409,  410,  410,  102,  411,  412,  413,
          414,   85,  415,  405,  416,   90,   11,   91,   13,   92,   16,   35,
           18,   26,   79,   92,   16,   35,   18,   24,   93,   91,   35,   10,
           29,   26,   17,   26,   94,   26,   85,   93,   13,   92,   16,   35,
           18,   79,   92,   16,   35,   18,   35,  418,  419,  420,  421,   81,
          422,  402,  403,  423,  424,   92,   16,   35,   10,   29,   18,   15,
           16,   85,   18,   13,   92,   16,   35,   18,   10,  351,  424,   92,
           16,   35,   10,   29,   18,   15,   16,   85,   18,   79,   92,   16,
           35,   18,   10,  318,  424,   92,   16,   35,   10,   29,   18,   15,
           16,   85,   18,   11,   91,   13,   92,    0],
        [   7,  113,   91,   13,   93, 1423,   91,   19,  887,   26,   34,  887,
           93, 1066,  113,   91,   29,   19,   13,   93,   10,  113,   91,   29,
           34,   13,   93,  113,   91,   29,   93,   10,   17,  113,   91,   29,
           93,   34,  113,   91,   17,   93,   34,  113,   91,   20,   93,   34,
          721,   34,  113,   91,  203,   93,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 800,   50,   59,  713,  182,  520,  231,  800, 1069,   68,  464,  691,
         1070,    9,  800, 1069,   68,  465,  258, 1071, 1070,   28,  800,   55,
          464,  465, 1072,  102, 1073, 1072,  110,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  660, 1844,  234,   62,  523, 1845, 1844,   56,  151,   64,   62,
           66,  256,  135,  104,   62,   68,  145, 1846, 1847, 1848,  154, 1849,
          153,   57,  957,  669, 1850, 1851,  154, 1849,  153, 1083, 1084,  145,
           83,   16, 1846,   18,   26,   27,  145,   83,   16, 1852,   18,   26,
           27,  145,   83,   16, 1853,   18,   26,   27,   83,   16,   66,   18,
          256,  135,  455, 1854, 1855,   27,  145,   83,   16,  151,   18,   51,
         1058,  249,  877, 1854, 1855,  247,   81, 1019,   15,   20,  168,  179,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,  113,   91,   13,   93,   10,  886,   91,   17,   13,   34,  116,
           16,  117,   18,   16,   20,   18,   93,  122,  580,  123,   91,   27,
           93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  446, 1472,  829, 1473,  829,  764, 1474,  829, 1475,   81, 1476,
         1477, 1062,  102, 1063,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  116,   16,   29,   18,   16,
           13,   18,   19,   13,   34,  125,  174,   13,  250,  113,   91,   13,
           93,  251,  252,  113,   91,   13,   93,  138, 1113,   68,   13,   92,
           16,   29,   18,  584,   13,   92,   16,   17,   18,  241,  116,   16,
          113,   11,   91,   13,   92,   16,   29,   18,   24,   93,   19,  113,
           11,   91,   13,   92,   16,   17,   18,   24,   93,   18,   16,   13,
           92,   16,   29,   18,   19,   13,   92,   16,   17,   18,   18,   22,
          125,   19,   17,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  180,   11,   12,  125,   92,   16,  131,   18,   24,   25,   11,
           12,  170,   92,   16,  131,   18,   24,   25,  125,   92,   16,   29,
           18,   10,   29,   26,   27,  170,   92,   16,   29,   18,   10,   23,
           21,  125,   92,   16,  131,   34,   29,   18,   10,   20,  125,   92,
           16,  131,   18,   19,  170,   92,   16,  131,   18,   34,   21,   26,
           27,   21,  170,   92,   16,  131,   34,   29,   18,   10,   20,  170,
           92,   16,  131,   18,   19,  125,   92,   16,  131,   18,   19,   21,
          241,   11,   12,  125,   92,   16,  131,   18,   34,  170,   92,   16,
          131,   18,   24,   25,  132,   11,   12,  125,   92,   16,  131,   18,
           19,  170,   92,   16,  131,   18,   24,   25,  370,  229,   11,   12,
          125,   92,   16,  131,   18,   24,   25,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 274,  275,  276,  102,  107,   65,   92,   29,  726,   13,   10,   19,
           17,  104,   65,   92,   16,   17,   18,  137,   91,   13,   19,   29,
           93,   15,   16,   17,   18,   34,   91,   79,   19,   17,   93,   15,
           16,   17,   18,   10,   29,  140,  141,  280,   13,  456,  580,  282,
          283,  284,  285,  272,  229,   65,   92,   29,  584,   65,   92,   17,
          286,  178,  107,   65,   92,   20,  286,  178,  288,   10,  116,   16,
          117,   18,   16,   21,   18,   91,  287,  294,  573,   93,   61,   65,
           92,   17,   65,   92,   20,  267,  464,  465,  229,  145,   65,   92,
           17,  470,   53,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1816, 1817, 1818, 1819, 1820, 1821, 1822, 1823, 1824, 1825,   74,  766,
         1826, 1827, 1828, 1829, 1830, 1831, 1832,  710, 1833, 1831, 1832,  766,
         1784,    7,  180,  182,  837, 1834,   17,   15,   16,   23,   18, 1835,
           17,   15,   16,   23,   18,   26,   17,   15,   16,   29,   18, 1836,
           17,   15,   16,   23,   18,   26,   17,   15,   16,   29,   18,   26,
           17,   15,   16,   17,   18, 1837,  229,   86,  122, 1546,  465,  465,
          216,  180,  181,  465,  183, 1546, 1838, 1839, 1831, 1832,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  370,   11,   12,  125,   92,   16,  131,   18,   24,   25,  181,
          131,  183,   83,   16,  353,   18,   92,   16,  131,   18,  132,   11,
           12,  170,   92,   16,  131,   18,   24,   25,  181,  131,  183,   83,
           16,  499,   18,   92,   16,  131,   18,  125,   92,   16,   29,   18,
           10,   19,   29,  170,   92,   16,   29,   18,   10,   29,  125,   92,
           16,   17,   18,   34,  170,   92,   16,   17,   18,   10,   17,  125,
           92,   16,   20,   18,   34,  170,   92,   16,   20,   18,   10,   30,
          229,   11,   12,  170,   92,   16,  131,   18,   24,   25,  526,  527,
           83,   16,  499,   18,   92,   16,   20,   18,   10, 1598,  229,   83,
           16,  353,   18,   92,   16,   20,   18,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  37,   38,   39,   40,   41,   42,   43,   44,   45,   46,   47,   48,
           49,   47,   50,   51,   52,   53,   54,   49,   47,   55,   56,   53,
           55,   56,   57,   58,   59,   52,   51,   60,    3,    3,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,   62,  140,  141,  265,  464,  255,   65,  137,  116,   16,   13,
           15,   16,   17,   18,   18,   16,   17,   18,   34,   79,   15,   16,
           17,   18,   10,   29,  464,  585,   13,  456, 1221, 1571,  465,   68,
          142,  261,   16,   83,   16, 1615,   18,   18,   10,  228,   16,   17,
           18,  261,   16,   83,   16, 1616,   18,   18, 1180,  142,  512,  178,
         1270,  568,  107,   13,   10,   19,   20,  261,   16,   83,   16,  144,
           18,   18,  263,  261,   16,   83,   16, 1256,   18,   18,   10,   29,
          241,  106,  142,  591, 1617,  107,  277,   65,  258,  139,  135,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 230,   13,   10,   19,  116,   16,   17,   18,   16,   20,   18,   26,
          114,   17,   83,   16,   13,   18,   10,    4,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  32,   91,   29,   19,   35,   93,   15,   16,   21,   18,   10,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 185,  186,  187,  188,  189,  188,  190,   81,  191,  192,  193,    9,
           26,   28,   26,   65,   26,   66,   99,  194,  188,  189,  195,    9,
          196,   28,  196,   65,  196,  197,  198,  199,  200,  201,  202,  203,
          202,   85,  202,   66,  196,  197,  198,  204,  205,  206,  203,  202,
          207,  208,  209,  138,  210,  211,  188,  189,  212,  210,  188,  213,
          214,  202,  191,  209,  210,  188,  213,   85,  202,  191,  197,  210,
          211,  188,  189,  138,  210,  215,  188,  216,  191,  190,  217,  190,
          194,  218,  219,  212,  209,  220,  210,  188,  221,  190,    9,  196,
           71,  222,  209,  220,  210,  188,  216,  223,  224,  224,  197,  225,
          210,  211,  188,  189,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1333, 1654, 1655, 1153,  212,  209, 1537, 1582, 1252,  212, 1582, 1656,
          209, 1543, 1582,  212,   58, 1543, 1582, 1656, 1582,  495, 1657, 1582,
         1658, 1252, 1063, 1657, 1582, 1659, 1500, 1660,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   11,   12,  125,   92,   16,  131,   18,   24,   25,  599,  132,
          125,   92,   16,   29,   18,  125,   92,   16,   17,   18,   26,   27,
          125,   92,   16,   20,   18,  600,  601,  229,   11,   12,  125,   92,
           16,  131,   18,   24,   25,  599,  125,   92,   16,   29,   18,   10,
           29,   26,  229,  180,   11,   12,  602,   92,   16,  131,   18,   24,
           25,  181,  131,  183,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   16,   13,   36, 1760,  822, 1761,   28,   16,   13,
           36, 1762,  822,    9,  760,   28,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 688,  520,  798,  691,  258, 1071,   52, 1809,  395,   52,   51,  692,
         1809,  395,  688, 1810, 1811, 1812, 1812,   53,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  125,   26,  170,   26,  162,  294,  573,   26,  125,   34,  170,
           34,  162,   10,   23,   26,  574,   10,   29,  241,  575,   34,  540,
           34,  576,   22,   23,  577,   12,  125,   26,  170,   26,  162,   25,
          125,   26,  170,   26,  162,  102,  168,  241,  577,   12,  125,   26,
          170,   26,  162,   25,  165,  228,  118,   20,  119,   16,   21,   18,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 104,   13,   15,   16,   17,   18,   34,   79,   15,   16,   17,   18,
           19,   17,   13,   19,  134,   79,   34,  729,   10,   23,  234,  107,
          641,   34,   79,   19,   29,   10,   23,  358,  125,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   16,  822,  822,   28,   16,   13,   36,   13,  822,
           13,    9,   31,   28,   10,   91,   27,   93,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1145,  993,  705, 1146, 1147,  528, 1148,  837, 1149, 1146,  102, 1150,
          413, 1151,  414,  216,  705,  528, 1152, 1153, 1154, 1155,  528, 1156,
          705,  528,  102, 1156,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1517, 1518, 1519, 1520, 1521, 1522, 1520, 1523,   97, 1524, 1525, 1526,
         1527, 1517, 1520,  348, 1528,  483, 1524, 1521,   71,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1011,  406,  924,  175, 1012,  893, 1013, 1014, 1012,  530,  414,  182,
          924, 1017, 1018,   81, 1019, 1860,  893, 1299, 1034, 1012, 1861,  893,
          924, 1018, 1862, 1863,  465, 1864,  944,  893, 1861,  301,  973,  414,
          182,  924, 1865, 1018,  822, 1864, 1864, 1033,  924,  433,  229,  142,
          973, 1661,  973,  766, 1866, 1032,  924,  102, 1018,  822, 1864, 1864,
         1033,  924, 1034, 1012,  893, 1013, 1035, 1036,  534,  893, 1013, 1037,
         1041, 1011,  893, 1013,  413, 1867, 1014,  414,  182,  924, 1018,  423,
          425,   16,   13,   18,   10,  116,   16,   29,   18,   16,  182,   18,
          424,   92,   16,   35,   10,   29,   18,   15,   16,  182,   18,   13,
           92,   16,   35,   18,   10, 1020,   26,   27, 1021,   10,  228,   16,
          116,   16,   29,   18,   16,  182,   18,    0],
        [  61,   33,   10,  116,   16,   20,   19,   35,   18,   16,   29,   34,
           17,   35,   18,   36,   33,   36,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 104,   13,   15,   16,   17,   18,   34,   79,   15,   16,   17,   18,
           19,   17,   13,   19,  134,   79,   34,  729,   10,   23,  234,  107,
           83,   16,  125,   18,   13,   34,   79,   19,   29,   10,   23,  358,
           83,   16,  125,   18,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  800,   50,  138,   59, 1283,  604, 1284,  157,  148,  800,  247,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  125,
           92,   16,   29,   18,   34,  125,   92,   16,   17,   18,   10,   19,
           29,   26,   27,  125,   92,   16,   29,   18,   19,  125,   92,   16,
           20,   18,   10,   19,   20,   26,   27,  125,   92,   16,   21,   18,
           10,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  107,   65,   92,   16,   29,   18,  137,   27,
           13,   10,   19,   17,   26,   27,  104,   65,   92,   16,   17,   18,
          137,   91,   13,   19,   29,   93,   15,   16,   17,   18,   34,   91,
           79,   19,   17,   93,   15,   16,   17,   18,   10,   29,  140,  141,
          280,   13,  456,  580,  282,  283,  284,  285,  272,  229,   83,   16,
           65,   18,   92,   16,   29,   18,   26,   83,   16,   65,   18,   92,
           16,   17,   18,  286,  178,  107,   83,   16,   65,   18,   92,   16,
           20,   18,  286,  178,  288,   10,  116,   16,  117,   18,   16,   21,
           18,   91,  287,  294,  573,   93,   61,   83,   16,   65,   18,   92,
           16,   17,   18,   83,   16,   65,   18,   92,   16,   20,   18,  267,
          464,  465,  229,  145,   83,   16,   65,    0],
        [   7,  112,  113,   91,   13,   93,   10,  230,   13,   34,  116,   16,
           29,   18,   16,  230,   13,   18,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  353,   92,   16,  131,   18,  132,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,  125,   92,   16,   29,   18,
           10,  116,   16,   29,   18,   16,   20,   18,  584,  125,   92,   16,
           21,   18,   15,   16,   17,   18,   10,  125,   92,   16,  105,   18,
           26,  353,   92,   16,   30,   18,   10,  179,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738,    7,  173,   65,  137,   79,   10,  116,   16,   13,   15,   16,
           17,   18,   18,   16,   17,   18,   26,   66,  107,   79,   10,   19,
          116,   16,   29,   18,   16,   17,   18,  265,   66,  585,   65,  359,
          176,  586,    9,   26,   28,  241,  107,  153,  270,  256,   11,   91,
           23,   26,  116,   16,   30,   18,   16,   17,   18,   24,   93,  234,
          104,  107,  153,  357,  586, 1089,  153,  466,  229,  811, 2037,   53,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 369,  368,   17,   15,   16,   13,   18,   19,   17,   15,   16,   79,
           18,   22,   20,   15,   16,   19,   13,   18,   19,   20,   15,   16,
           19,   79,   18,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 136,  162,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
           21,   18,   19,  116,   16,   79,   15,   16,   17,   18,   18,   16,
           17,   18,   10,   29,  259,  139,  135,   68,  142,   65,  175,  360,
           62,  140,  141,   36,  605,   36,   10,   36,  146,   36,  145, 2031,
           53,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,   49,   47,  142,  822,   63,  102,  266,  172,   59,   63,  594,
          469,  469,  154,  153,   10,  594,   10,  155,   10,   20,  266,   10,
          154,   10,   21,  464, 1089,  594,  238,  467,   10,   17, 1254,  465,
          608,  466,  241,  470,  469,  469,  242,  243,  229,  915,  465,  822,
         1255,  247,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1177, 1178,   63,   19,    9,   92,   16,   29,   18,   28,   92,
           16,   29,   18,   65,   92,   16,   29,   18,   66,   92,   16,   29,
           18,   59,  812,  468,   92,   16,   29,   18,   10,   21,  584,  153,
           10,   17,  584,  239, 1179,   10,  351,   15,   16,   82,   18,  584,
          256,  584,  464,  465,  154,  584,  462,   92,   16,   29,   18,   26,
          584,    9,   92,   16,   29,   18,   66,  466,  241,  470,  469,  469,
          242,   65,   92,   16,   29,   18,  560,  229,  607,    9,   19, 1183,
           92,   16,   29,   18,   19,  465,  720,  610,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1412,  151,  102,   28,   10,  116,   16,  117,   18,   16,   21,   18,
          154, 1067, 1212,  116,   16,   29,   18,   16,   20,   18,  154,  114,
            9,   54,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   36,   13,   34,   29,   36,
           19,   36,   13,   19,   17,   36,  229,  290,  113,  990,   13,  991,
          589,   29,  293,  290,  113,   91,   13,   93,  589,   13,   15,   16,
           17,   18,   19,   13,   34,  171,  293, 1319,  229,  171,  254,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 980,   40,  423, 1529,  610, 1530, 1478,  231, 1478,  129,  127,  129,
          355,   13,  131,  300,  355,  125,  130, 1021,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   13,   15,   16,   17,   18,
           19,   17,   13,   34,  125,   11,   91,  248,   15,   16,   13,  263,
           29,   18,   34,  248,   15,   16,   19,   13,   34,   29,   18,   24,
           93, 1181,  253,  125,   10,   91,   27,   93,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  169,  889,   16,  125,   18,   10,   91,   13,   26,   13,   34,
           29,   93,  889,   16,  170,   18,   10,   91,   29,   26,   17,   93,
          889,   16,  125,   18,  172,  889,   16,  170,   18,   13,   10,    4,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   12,   13,   36,   13,  257,   19,   29,   25,
           28,   10,   12,   13,   36,   13,   22,   17,   25,    9,  760,   28,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,  116,   16,  641,   15,   16,
           17,   18,   34,   13,   19,   29,   18,   16,  248,   15,   16,   13,
           18,   18,  229,  173,   79,   10,  113,   91,   13,   93,   68,  822,
          579,  176,  178,  241,  125,  165,   29,  249,  113,   91,   13,   93,
           34,  248,  165,   23,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  173,   79,   10,   13,   15,   17,   34, 1312,
           19,   17,   13, 1220,    9,   28,  362,   68,   65,  140,   91,   23,
           26,   29,   93,  171, 1058,  249, 1313,  155,  172,  154, 1059,  449,
          241,    9,   28,   65,  379,  104,   79, 1314,  108,  109,  525, 1315,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 705, 1394,  705, 1395,  102,  869,  330, 1620, 1621, 1395,   71,   91,
           27,   93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [2173,  682, 1409, 2174,  446, 2175, 2176, 1497, 2177, 2175, 1582,  249,
         2176, 1497,  102, 2178, 2179, 2180,  417, 1497, 1497, 2176, 1497, 2181,
          733, 2182,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 127, 1414,  129,  355, 1415,  436,  130,  610,   54, 1021,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  125,
           92,   16,   29,   18,   10,   20,   26,   27,  125,   92,   16,   29,
           18,   34,  125,   92,   16,   20,   18,   34,  125,   92,   16,   30,
           18,   10, 1598,  125,   92,   16,   20,   18,   34,  125,   92,   16,
           30,   18,   34,  125,   92,   16,  133,   18,   10,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1338,  102, 1339, 1340, 1341,   71, 1342, 1343, 1341, 1344, 1341,   71,
         1345, 1346, 1341,   71,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   13,   15,   16,   20,   18,
           34,  752,   34,  162,   26,   13,  294,  573,  173,  113,   91,   13,
           93,   68,   11,   91,  116,   16,   29,   18,   16,   17,   18,   26,
          113,   11,   91,  116,   16,   29,   18,   16,   17,   18,   24,   93,
           24,   93,  579,  176,   79,  456,  352,  229,  170,  113,   91,   13,
           93,  753,  754,  253,  241,  113,   91,   13,   93,  253,  754,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  230,  288,   34,  230,   11,   91,  288,   34,  116,   16,  117,
           18,   16,   20,   18,   24,   93,   10,   29,  230,   11,   91,  288,
           34,  116,   16,  117,   18,   16,  105,   18,   24,   93,   10,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  102,  797,  112,   79, 1347,  797,  107,   13,  736,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  35,   91,   17,   34,   20,   35,   93,   10,   91,   27,   93,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  135,   92,   16,   29,   18,   26,  135,   92,   16,   17,   18,
          255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
         1426,   18,   34,  116,   16,   79,   15,   16,   17,   18,   18,   16,
           85,   18,   10,   29,  138,  139,  464,   65,  238,  670,  646,  363,
          716,   92,   16,   29,   18,  135,   92,   16,   17,   18,  957,  669,
          464,  140,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 800,  242,  801,  802,  803,  523,  451,  519,  519,  520,  102,  691,
          692,  798,  804,  519,  158,  182,   85,  805,  451,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  128,  129,  130,  131,    3,    3,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1601, 1602, 1603,  338, 1604, 1605, 1606,  413, 1607, 1608,  249,  530,
          414,  216, 1605, 1017, 1609, 1603, 1607,   81,  898,  942, 1101, 1103,
          301,    9, 1455, 1604, 1606, 1610, 1611, 1607, 1301, 1612,  219,    9,
           71, 1613,  553,  553,  554, 1382,  556, 1611, 1607, 1606,  413, 1611,
         1607,  942, 1101, 1103,  894, 1606,  413, 1614,  450,  503,   15,   16,
           17,   18,   10,  116,   16,  131,   91,  539,   19,  540,   93,   15,
           16,   17,   18,   18,   16,   91,  125,   34,  170,   93,   91,  162,
           34,  541,   93,   91,  125,   34,  162,   93,   91,  170,   34,  541,
           93,   18,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   10,   91,  171,   26,   21,   93,
          889,   16,  170,   18,   10,   91,   20,   26,   19,   17,   93,  261,
           16,   83,   16,  125,   18,   18,  469,  469,  261,   16,   83,   16,
          170,   18,   18,  171, 1533,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 824, 1351, 1352, 1351, 1172, 1174, 1353,  440, 1351, 1174,  824, 1354,
         1355,  448,  413,  411,  412, 1356, 1150, 1357,  448,  413, 1350,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   19,   13,   15,   16,   17,
           18,   34,  641,   34,   21,   26,  477,   91,   13,   93,   10,   36,
           13,   34,   29,   36,   34,   36,   13,   19,   29,   36,  125,  249,
          229,  290,  113,   91,   13,   93,  589,  477,   91,   13,   93,  293,
          290,  113,   91,   13,   93,  589,  477,   91,   13,   93,  293, 1049,
          229,  125,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  557,   63,   19,    9,   92,   16,   29,   18,   28,   92,   16,
           29,   18,   65,   92,   16,   29,   18,   66,   92,   16,   29,   18,
          102,   68,  256,   26,  135,  558,  559,   92,   16,   29,   18,   26,
          462,   92,   16,   29,   18,   17,  560,   10,  561,   92,   16,   29,
           18,   26,  562,   10,   17,  563,   92,   16,   29,   18,  241,  153,
           10,  154,  249,  564,  172,  155,   68,   65,   92,   16,   29,   18,
          242,  565,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    9,   28,  157,   62,  148,  362,  239,   83,   16,  958,   18,
           10,  201,   15,   16,   82,   18,   65,  148,  265,  246,   62,  151,
          247,  168, 1426,  157,   62,  158,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,   13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   20,
           13,   34,   17,   79,   19,  105,  163,   23,  164,   13,  165,   23,
          164,   79,  165,   23,  166,  161,   24,  167,   33,   10,   13,   19,
           79,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  174,   91,   29,   34,   36,
           13,   36,   93,   19,  116,   16,   29,   18,   16,   29,   34,   13,
           15,   16,   17,   18,   18,   26,  113,   91,   13,   93,  257,  113,
           91,   17,   13,   19,   29,   93,  698,   13,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 486,  416,   90,  102,   71,  373,   92,   16,   29,   18,   26,  373,
           92,   16,   17,   18,   26,  373,   92,   16,   20,   18,   26,  373,
           92,   16,   21,   18,  424,   92,   16,   35,   10,   29,   18,   15,
           16,   21,   18,  373,   92,   16,   35,   18,   10,   29,  723,  724,
          102,  416,  725,  486,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   51,   52,   63,  242, 1383, 1384, 1245,   16,  269,   18,  242,
          352,  464, 1245,   16,  269,   18, 1385,   65,   66,   68,  241,  242,
         1386,  172,  242, 1387,  246,  464,  822,  151,  247,  249, 2027, 2028,
         2029, 2030, 1074,  607,  720,  610,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  32,   33,   10,   29,   34,   35,   11,   36,   33,   15,   16,   17,
           18,   19,   17,   33,   24,   36,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 180,   11,   12,  125,   92,   16,  131,   18,   24,   25,  102,  125,
           92,   16,   29,   18,   10,   17,  125,   92,   16,  171,   34,  131,
           18,   10,  125,   92,   16,  171,   18,  125,   92,   16,  131,   18,
          125,   92,   16,  278,   34,   29,   18,   34,  125,   92,   16,  278,
           34,   17,   18,   34,   94,   34,  125,   92,   16,  278,   34,   96,
           18,   10,   17,   15,   16,  483,   18,   19,   17,   15,   16,   30,
           18,  278,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  811,   63,   52,  256,  135,  594,  154,  466, 1570, 1083,  145,
           83,   16, 1905,   18, 1084, 1085,   65, 1086,   68,  142, 1087, 1906,
          146,  172,  562,  241,  242, 1907,  172,  242, 1908,  229, 1909,  242,
         1908,  719,  720,  610,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,    7,  580,  246,  142,   19,  151,   55,  395,   83,   16,  266,
           18,   10,  105,  233,  142,  242,  151,  907,  908,   66,   66,  242,
          243,  907,  908,  256,  909,  910,  911,  361,  153,  853,  260,  241,
          260,  153,  466,  121,  102,  912,   68,  256,  242,  244,  907,  135,
          913,  449,  914,  915,  916,  247,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 712,  705,  328,  183,  331, 1727,  348,  183,  183,  331,  712, 2023,
          682,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738, 1439,  209,  894, 1440, 1441, 1442, 1443,  975, 1444,  409, 1441,
          530,  407,    9,   26,   28, 1369, 1370,  216,    9, 1418, 1441, 1445,
         1440, 1446,   28, 1418, 1441, 1447, 1440, 1446, 1441, 1448, 1446,  247,
         1449, 1450, 1451,  406, 1452,  413, 1453,  794, 1443, 1441, 1442, 1440,
         1454, 1444,   90, 1103,  301,   65, 1455,  209, 1440, 1443, 1442, 1454,
         1301, 1456, 1103,  142,   91,   65,   93,  430, 1457, 1458, 1440, 1443,
         1454, 1103,  102,  125,   26,  170,  610,  219,  212,  209, 1440, 1443,
         1454, 1459,  486,  102,   90,  545,  546,  547,  548,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   17,  174,   13,   34,   29,
          113,   91,   13,   93,  163,   17,   13,   34,  162,   26,  229,  162,
          254,  476,  125,  257,   23,   26,  250,  112,  477,   91,   13,   93,
           10,  116,   16,  113,   91,   13,   93,   19,  113,   91,  125,   93,
           18,   16,   13,   19,  125,   18,  251,  252,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   49,   47,  142,   19,   63,  102,  153,  469,  469,  269,  239,
         1005,   10,  239,   16, 1006,   18,   10,  201,   15,   16,   82,   18,
          241,  242,  243,  172,  242, 1007,   83,   16,  266,   18,   10,   83,
           16, 1008,   18,   10,   83,   16,  153,   18,   10,   83,   16, 1009,
           18,   26,   27,  239,   83,   16, 1010,   18,   10,  201,   15,   16,
           82,   18,   49,   47,  142,   19,   63,  247,  116,   16,  134,   18,
           16,   20,   18,  429,   49,   47,  245,   53,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1188, 1189, 1190, 1191, 1192, 1059, 1193, 1050,  102, 1194,  544, 1192,
          544, 1195, 1196,  121,  593,    9,   68, 1197,  544, 1192,  510,  483,
           28,   68, 1198,  544, 1195,  510, 1199, 1063,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,   11,   12,  125,   92,   16,  131,   18,   24,   25,  132,  125,
           92,   16,   29,   18,   34,  125,   92,   16,   17,   18,   34,  125,
           92,   16,   20,   18,   10,   29,   26,  125,   92,   16,   17,   18,
           34,  125,   92,   16,   20,   18,   34,  125,   92,   16,   21,   18,
           10,   17,   26,  125,   92,   16,  105,   18,   34,  125,   92,   16,
          133,   18,   34,  125,   92,   16,  134,   18,   10,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  135,   92,   16,   29,   18,   26,  135,   92,   16,   17,   18,
          255,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
         1426,   18,   34,  116,   16,   79,   15,   16,   17,   18,   18,   16,
           85,   18,   10,   29,  138,  139,  464,   65,  238,  670,  646,  363,
          716,   92,   16,   29,   18,  135,   92,   16,   17,   18,  957,  669,
          464,  140,    4,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   13,   15,   16,   20,   18,
           34,  641,   34,  116,   16,   29,   18,   16,   21,   18,   26,  477,
           91,   13,   93,   10,   19,  174,   13,  125,  847,   13, 1757,  173,
           79,   10,  113,   91,   13,   93,  176, 1016,   16,  171,  131,  171,
          131,  102,  111,   61,  112, 1758,   91,   13,   93,   10, 1759,   12,
          113,   91,   13,   93,   26,  477,   91,   13,   93,   25,   91,   13,
         1724,   23,   93,  250, 1758,   91,   13,   93,  253,  307,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  513,   65,   79,   15,   17,   10,   17,   13,  139,  135,  382,
           13,  456,  359,  107,  277,   92,   29,   26,  277,   92,   17,  361,
           65,    9,   28,  362,  361,   65,  458,  142,  568,  362,  135, 1089,
          153,  573, 1256,  466,  241, 1257,  469,  469, 1258,  145, 1259,   53,
          145, 1260,   53, 1261,  229,  153,  466,  512,  178,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   79,  159,   11,   12,  160,  161,   16,  277,   18,   13,   34,
           79,   19,   17,  578,   23,  164,   13,   19,   17,   79,   34,   29,
          163,   23,  164,   17,   13,   19,   79,   34,   17,  589,   23,  166,
          161,   24,  167,   33,  807,   79,  168,  179,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16, 1187,   18,   10,   91,  116,   16,   29,   18,
           16,   17,   18,   26,  116,   16,  228,   16,   20,   18,   18,   16,
           17,   18,   93,   26,  889,   16,  154,   18,   10,   91,  116,   16,
          228,   16,   20,   18,   18,   16,   17,   18,   26,  116,   16,   29,
           18,   16,   17,   18,   93,  239,  151,   10,    4,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  11,   91,   29,   34,   17,   13,   15,   16,   17,   18,   24,   93,
           91,   29,   34,   13,   93,   15,   16,   21,   18,  582,  102,   13,
           15,   16,   20,   18,  583,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[1575, 1309,  169,  889,   16,  125,   18,  889,   16,  170,   18,   36,
          889,   16,  125,   18,   34,  889,   16,  170,   18,   36,   10,   36,
          889,   16,  125,   18,   19,  889,   16,  170,   18,   36,   91,   27,
           93,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   83,   16,  135,   18,   92,   16,   29,   18,   26,   83,   16,
          135,   18,   92,   16,   17,   18,  136,  256,  137,  116,   16,   13,
           15,   16,   17,   18,   18,   16,   83,   16,  125,   18,   15,   16,
           17,   18,   18,   19,  116,   16,   79,   15,   16,   17,   18,   18,
           16,  170,   15,   16,   17,   18,   18,   10,   29,  258,  259,  139,
           68,  464,  256,  716,   92,   16,   29,   18,   13,  456,  352,  230,
          239,  716,   92,   16,   17,   18,  135,   92,   16,   29,   18,   10,
          116,   16,   29,   18,   16,   20,   18,  256,  457,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  486,  416,   90,   13,   92,   16,   29,   18,   26,   13,   92,
           16,   17,   18,   26,   94,   26,   13,   92,   16,  131,   18,  487,
          436,   90,   96,   13,   92,   16,   29,   18,   26,   96,   13,   92,
           16,   17,   18,   26,   94,   26,   96,   13,   92,   16,  131,   18,
          487,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738,  121,  463, 1461,   26, 1462,  363,  151,  812, 1463,  242,  979,
          153,   10,   29,   26,  814,   10,  562,   10,   17,  239, 1464,   10,
          351,   15,   16,   82,   18,  153,   26,  154, 1084,  814,  562,  455,
          951, 1465,  231,  241,  121,  102,  960,    9,   26,   65,   26,  260,
           26,   66, 1116,  718,  242,  151,  172,  242, 1466, 1467,  102,  607,
           28, 2035, 2036, 2035,    9,  857,    3,    3,    3,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 112,  113,   91,   13,   93,   10,  114,   11,   91,   20,   13,   34,
          116,   16,  117,   18,   16,  105,   18,   24,   93,  118,   23,  584,
          117,  119,  253,  307,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 107,   13,   79,   13,  456,   79, 1220,    9,   28,  362,   68,  142,
          104,   13,   15,   16,   17,   18,   34,   79,   15,   16,   17,   18,
           10,   17,  145,   83,   16, 1091,   18,   53,  254,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 125,  170,  162,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 528,  529,  530,  405,  531,  216,  532,  533,  194,  534,  535,  536,
          537,  218,   90,  538,   81,  500,  219,  542,  533,  194,   71,  229,
          543,  535,  536,  544,  537,  430,  486,  102,   90,  545,  546,  740,
          741,  548,  549,  533,  194,  303,  550,  533,  549,  533,  194,  303,
          550,  533,  551,   90,  552,  553,  553,  554,  555,  556,  543,  535,
          536,  537,  542,  534,  533,  450,  503,   15,   16,   17,   18,   10,
          116,   16,  131,   91,  539,   19,  540,   93,   15,   16,   17,   18,
           18,   16,   91,  125,   34,  170,   93,   91,  162,   34,  541,   93,
           91,  125,   34,  162,   93,   91,  170,   34,  541,   93,   18,   26,
           27,  116,   16,  142,   11,   91,  503,   15,   16,   17,   18,  589,
          278,   24,   93,   18,   16,  278,   18,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 274,  275,  276,  102,  173,   83,   16,   65,   18,   92,   16,   29,
           18,  273,  178,   11,   12,  160,  161,   16,  277,   18,   13,   10,
          228,   16,   20,   18,  114,  384,  164,   79,   10,  230,  384,  166,
          161,   24,  167, 1393,  273,  140,  141,  280,   13,  456,  580,  282,
          283,  284,  285,  272,  173,   83,   16,   65,   18,   92,   16,   17,
           18,  286,  178,  885,   16,  287,  886,   18,   11,   91,  288,   34,
          116,   16,  117,   18,   16,   21,   18,   24,   93,   10,   17,  228,
           16,   17,   18, 1269,   83,   16,   65,   18,   92,   16,   29,   18,
          178,   83,   16,   65,   18,   92,   16,   17,   18,  274,  178, 1270,
          142,   83,   16,   65,   18,   92,   16,   29,   18,   68,  568,   83,
           16,   65,   18,   92,   16,   17,   18,    0],
        [  61,  513,   83,   16,   65,   18,  137,   27,   83,   16,   79,   18,
           15,   16,   17,   18,   10,   17,   83,   16,   13,   18,   68,    9,
           91,   17,   26,   23,   93,   26,   28,   91,   19,   17,   26,   23,
           93,  106,    9,  107,  277,   65,  515,  464,  465,  362,  502,   13,
          456,  352,  249,  229,  107, 1047,  178,  241,  239,   83,   16, 1110,
           18,   10,  239,   83,   16, 1111,   18,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1124, 1125, 1126, 1127, 1128, 1129,  110, 1129, 1130,  110,  116,   16,
          228,   16,   30,   18,   19,   29,   18,   16,   17,   18,  116,   16,
          228,   16,   30,   18,   19,   29,   18,   16,   17,   18,  452, 1131,
          662, 1132, 1060, 1133, 1134, 1135, 1136, 1126, 1127, 1128, 1137,  110,
         1137, 1129,  110,  116,   16,  228,   16,   30,   18,   19,   29,   18,
           16,   17,   18, 1138,  138, 1132, 1060, 1139,  525, 1140, 1128, 1141,
         1142,  110, 1143, 1019, 1144,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   15,   16,  522,   18,   91,   13,   93, 1066,  113,
           91,   13,   93,   27,   91,   13,  294,  573,   93,  850,  112,  113,
           91,   19,   29,   93,   10,   23,   13,  962,   23,  249, 2039,   15,
           16,  522,   18,   91,   13,   93,   19,  113,   91,   13,   93,   22,
           23,  113,   91,   13,   93,  257,   23,  698,   13,  254,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  112,  113,   91,   13,   93,   10,   13,   15,   16,   20,   18,
           34,   91,  125,   19,   29,   93,   13,   15,   16,   17,   18,   34,
          641,  113,   91,   13,   93, 1066,  173,   79,   10,  113,   91,   13,
           93,   68,  579,  176,  178,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 371,   99,  372,  373,   92,   16,   29,   18,  220,  220,  374,  238,
          375,  107,  376,  242,  373,   92,   16,   17,   18,  377,  102,  378,
          379,  380,  381,  242,  373,   92,   16,   20,   18,  377,  359,  107,
          374,  359,  107,  382,  373,   92,   16,   21,   18,  107,  277,  383,
          242,  384,  107,  171,  172,  242,  384,  171,  172,  277,  372,  102,
          386,  372,  387,    4,  373,   92,   16,   29,   18,  389,  373,   92,
           16,   21,   18,  373,   92,   16,   29,   18,  389,  373,   92,   16,
           17,   18,  392,  373,   92,   16,   17,   18,  393,  373,   92,   16,
           20,   18,  392,  373,   92,   16,   20,   18,  393,  392,  373,   92,
           16,   21,   18,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   13,   15,   16,   20,   18,
           19,  566,   34,  278,   15,   16,   17,   18,  250,  113,   91,   13,
           93,  251,  252,  113,   91,   13,   93,  149,  253,  229,  278,  254,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   12,   13,   36,   13,   19,   29,  165,   23,
           25,   83,   16,   28,   18,   10,   12,   23,   26,   27,   29,   26,
           27,   17,   25,    9,   31,   28,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 980, 1320, 1321, 1322, 1323, 1324, 1325, 1324, 1326, 1327, 1328, 1329,
         1330, 1324,  686,  750, 1327, 1331, 1332, 1323, 1324, 1333, 1329, 1323,
         1324, 1334, 1335,  750, 1336, 1337, 1329, 1323, 1324,  692,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  116,   16,  230,   13,   34,   13,
           18,   16,  114,   13,   34,   13,   15,   16,   17,   18,   18,  118,
           19,  117,   26,  117,  119,  120,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  177,  278,  107,  277,  255,   65,  116,   16,   13,   15,   16,
           17,   18,   18,   16,   21,   18,   34,  116,   16,   79,   15,   16,
           17,   18,   18,   16,   20,   18,   10,   29,  515,    9,   28,  362,
         1089,  153,  466,  464,  171,  171,  241,  278,   22,   19,  116,   16,
           29,   18,   16,   17,   18,   61,  135,   65,  259,  139,  142,   65,
          238,  261,   16,   83,   16, 1390,   18,   18,   34,  261,   16,   83,
           16, 1391,   18,   18,   34,  261,   16,   83,   16,  563,   18,   18,
           10,  261,   16,   23,   18,  241,   17,   36,  261,   16,   83,   16,
         1390,   18,   18,   36,   10,   36,  261,   16,   83,   16, 1391,   18,
           18,   36,   34,   36,  261,   16,   83,   16,  563,   18,   18,   36,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  296,   16,  151,   18,  226,    9,   28,   65,  125,  170,  162,
          145,  296,   16,  151,   18,   53,  116,   16,  125,   15,   16,   17,
           18,   34,  170,   15,   16,   17,   18,   19,  162,   15,   16,   17,
           18,   18,   16,   21,   18,   65,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 750,   83,   16,   63,   18,   19,   83,   16,    9,   18,   92,   16,
           29,   18,   83,   16,   28,   18,   92,   16,   29,   18,   83,   16,
           65,   18,   92,   16,   29,   18,   83,   16,   66,   18,   92,   16,
           29,   18,  102,   83,   16,  153,   18,   10,   83,   16,  154,   18,
           10,   29,   26,   27,   83,   16,  468,   18,   92,   16,   29,   18,
           10,  228,   16,   20,   18, 1625,  107,   83,   16,  594,   18,   92,
           16,   29,   18,   83,   16, 2135,   18,   92,   16,   29,   18,  719,
          609,  741,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  13,   26,   79,  730,   84,   11,   12,  160,  161,   16,  277,   18,
           13,   34,   79,  165,   23,  164,   17,   13,   19,   79,  165,   23,
          164,   13,  163,   29,  166,  161,   24,  167,   33,   10,   20,   13,
           34,   17,   79,  168,    4,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1189, 1190, 1275, 1276,  441, 1188, 1167, 1277,  218, 1225,  829, 1194,
         1253,  829,  298, 1194,  313, 1194, 1278, 1279,   81, 1280,   90, 1193,
         1230,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  129, 1478, 1479, 1480,  764,   40,  766,  767,  763,  102, 1481,
         1482, 1483,  127,  129,  355,  125,  170,  759, 1484,  130,  125,    3,
            3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  112,  113,   91,   13,   93,   10,  114,   11,   91,   13,   34,
          116,   16,  117,   18,   16,   20,   18,   24,   93, 1062, 1281,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  290,  291,    7,  113,   91,   13,   93,   10,   36,   13,   34,
           29,   36,   19,   36,  641,   19,   29,   36,  125,   10,   29,  249,
          229,  290,  113,   91,   13,   93,  257,   29,  293,   13,  294,   91,
           23,   26,   29,   93,  249,  290,  113,   91,   13,   93,  257,   13,
          229,  125,  254,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  173,   79,   10, 1112,   15,   16,   13,   18,   34,   13,  174,
           13,   68,   91,   29,   26, 1112,   93,  579,  176,  178,   79,   10,
           17,   13,   34,  170,    4,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 750, 1595, 1596,   58,  233,  157,   62,  148,  157,   62,  158,  179,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 301,  353,   92,   16,  131,   18,  370,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,    7,  125,   92,   16,   29,
           18,   10,   19,  133,   26,   27,  353,   92,   16,   20,   18,   10,
           19,  483,  229,   11,   12,  125,   92,   16,  131,   18,   24,   25,
          526,  527,  229,   83,   16,  353,   18,   92,   16,  131,   18,  914,
           83,   16,  353,   18,   92,   16,  131,   18,  111,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 230,  384,   10,  116,   16,   29,   18,   16,   20,   18,   26,  114,
           17,  384,   10,    4,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   11,   12,  160,  161,   16,
          277,   18,   17,   15,   16,   13,   19,   29,   18,   19,   17,   26,
           13,  578,   29,  164,   19,  126,   92,   16,   17,   18,   91,   13,
           34,   29,   93,   26,   13,  257,   29,  166,  161,   24,  167,  113,
           91,  125,   93,   10,   19,   20,  113,   91,  105,   19,  125,   93,
           10,   91,   27,   93,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  169,  889,   16,  125,   18,  889,   16,  170,   18,  382,  169,
         1348,  889,   16,  125,   18,   34,  889,   16,  170,   18,  889,   16,
          125,   18,   34,   17,  889,   16,  170,   18,  382,  861, 1348,  737,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 145,  151,  226,    9,   28,   65,  125,  170,  162,  114,    9,   10,
          116,   16,   21,   18,   16,   30,   18,  114,   65,   10,  116,   16,
           30,   18,   16,  729,   18,  125,  170,  737,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  72,   73,   74,   75,   76,   77,   87,   78,   79,   80,   13,   81,
           15,   16,   82,   18,   83,   16,   65,   18,   84,   85,   80,   86,
           87,   88,   89,   89,   90,   11,   91,   13,   92,   16,   35,   18,
           26,   79,   92,   16,   35,   18,   24,   93,   91,   35,   10,   29,
           26,   17,   26,   94,   26,   85,   93,   95,   95,   96,   15,   16,
           82,   18,   83,   16,   65,   18,   97,   15,   16,   82,   18,   83,
           16,   65,   18,   98,   99,  100,  101,  102,  103,   78,   79,   80,
           13,  100,  101,    3,    3,    3,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 274,  275,  276,  102,  107,  277,   92,   29,  273,  178,   11,   12,
          160,  161,   16,  277,   18,   13,   10,   17,   34,  279,  164,   79,
           10, 1317,  166,  161,   24,  167,  279,  273,  107,  277,   92,   17,
          273,  178,   11,   12,  160,  161,   16,  277,   18,   13,   10,   19,
           17,   34,  171,  164,   79,   10,  116,   16,  171,   18,   16,  278,
           18,  166,  161,   24,  167,  171,  273,   61,  277,   92,   29,  277,
           92,   17,  267,  142,  278, 1058,  249,  142,  512,  173,   65, 1269,
           65,  178,  140,  141,  280,   13,  281,  282,  283,  284,  285,  272,
           61,  277,   92,   16,   20,   18,  137,  287,   91,  114,  288,   34,
          230,  288,   93,   19,  228,   16,   17,   18,   10,   23,  464,  277,
           92,   20,   65,  267,  229,  464, 1318,    0],
        [ 764,   40, 1929, 1930, 1931,  700, 1932, 1058, 1933, 1934, 1935,  704,
         1936, 1936,  193, 1937, 1938,  231, 1939, 1931, 1934,  102,  530,  987,
         1939, 1931, 1934, 1652, 1940, 1936,   71,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  125,  257,   23,   26,   27,  170,  257,   23,   26,   27,  125,
           15,   16,   20,   18,   34,  170,   15,   16,   20,   18,   10,   17,
          241,   91,  125,   34,  170,   93,   11,   91,  125,   15,   16,   30,
           18,   34,  170,   15,   16,   30,   18,   24,   93,  589,   21,  125,
           34,  170,  163,   17,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 738,  363,  151,  226,    9,   26,   28,   26,   65,  125,   26,  170,
           26,  162,    7,  125,  230,  116,   16,    9,   34,   65,   18,   16,
           17,   18,   10,  170,  230,    9,  229,   28,  363,  151, 1460,  669,
          162,   10,   29,  229,  363,  151,   53,  254,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  174,   91,  228,   16,   29,
           34,   13,   15,   16,   17,   18,   18,   19,   13,   93,   34,   29,
          113,   91,  125,   93,   10,   21,  113,   91,   19,  125,   93, 1359,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 271,  290,  291,   61,   13,   26,   79,   26,   33,  294,  573,   13,
           34,   79,   34,   33,   10,   29,  229,   91,   13,   19,   29,   93,
           15,   16,   17,   18,   34,   91,   79,   34,   29,   93,   15,   16,
           17,   18,   34,   91,   33,   34,   29,   93,   15,   16,   17,   18,
          111,   91,   13,   19,   17,   93,   15,   16,   17,   18,   34,   91,
           79,   19,   29,   93,   15,   16,   17,   18,   34,   91,   33,   19,
          125,   93,   15,   16,   17,   18,  165,  116,   16,   29,   18,   16,
           20,   18,  698,  241,  125,  163,   19,   20,  125,  165,   19,   29,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  173, 1813,   79,   10,  114,   13,   26,   27,   65,   92,   16,
           17,   18,  137,   79,   10,  230,   11,   91,   17,   13,   34,  116,
           16,   17,  117,   18,   16,   20,   18,   24,   93, 1062, 1063,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  255,   65,  116,   16,   13,   15,   16,   17,   18,   18,   16,
           17,   18,   34,   79,   15,   16,   17,   18,   10,   29,  259,  139,
          135,  135,  107,  277,   65,  515,    9,   28,  362,   68,  464,  140,
          277,   13,  456,  352,  249,  229,  107,  467,  178,   61,   62,  140,
          141,  241,  239,   83,   16, 1910,   18,   10,  239,   83,   16, 1911,
           18,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,  248,   15,   16,   13,   18,
           11,   91,  248,   15,   16,   13,   18,   19,  125,   24,   93,   19,
          125,   15,   16,   17,   18,   13,  250,  113,   91,   13,   93,  251,
          252,  113,   91,   13,   93,  589,   23,  229,  125,  254,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,   68,    9, 2035,   28,  265,  464,   13,   79,  107,  467, 1047,
          177, 2207,   19,  116,   16,   29,   18,   16,   17,   18,  301,  464,
          512,  173,   65,  229,   65,  178,   65,  173,  140,  141,  107,  361,
           65,  142,  568,  362,   68,  142,  670,  646,  910,  815,   13,  456,
         1571,  256,  951, 2208,  911,  361,   65,  853,  260,   35,  241,  145,
         2209,  395,  846,  229,  145, 2209,   53,  168,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,  750,   63,  822, 1501,  102,  153,  182,  154,   96, 1502,   68,
          256,  135,   83,   16,    9,   18,   92,   16,   29,   18,   83,   16,
           28,   18,   92,   16,   29,   18,   26,   27,   83,   16,   66,   18,
           92,   16,   29,   18,   83,   16,   65,   18,   92,   16,   29,   18,
           26,   27,   83,   16,    9,   18,   92,   16,   29,   18,   83,   16,
          256,   18,   10,   83,   16,   66,   18,   92,   16,   29,   18,   83,
           16,  135,   18,   10,   21,  167,   27,   83,   16,  256,   18,   26,
           27,  135, 1503,  384,  750,  718,  374, 1504,  858,   52,  121,  964,
          794,   52, 1505, 1506,  449,  229,  242,  384,  750,  407,  220,  247,
           60,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231, 1859,  142,   63,  102,  153,  269,  239, 1005,   10,  239, 1006,
           10,  201,   15,   16,   82,   18,  241,  242,  243,  172,  242, 1007,
          266, 1008,  153, 1009,  239,   83,   16, 1010,   18,   10,  201,   15,
           16,   82,   18,  229,  607,    9,  268,   65,  609,  610,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1412,  151,  102,  114,  116,   16,   65,   18,   16,   17,   18,   10,
          116,   16,  228,   16,   30,   18,   18,   16,   30,   18,  154,  155,
          153,   91,   27,   93,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   11,   12,   91,   13,   26,   79,   93,   36,
           13,   15,   16,   17,   18,   34,   79,   15,   16,   17,   18,   10,
           29,   24,   25,   26,   27,   28,   10,   12,   91,   13,   26,   79,
           93,   36,   79,   10,   13,   25,    9,   31,   28,  102,  484,  307,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    8,    9,   16,   13,   36,   13,  757,  131,  758,  465,   28,
           16,   96,  298,  759,    8,    9,  760,   28,  102,  484,  307,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  104,   13,   15,   16,   17,   18,   34,   79,   15,   16,   17,
           18,   19,  105,   13,   10,   23,  106,  107,  104,  108,  109,  110,
          111,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  446, 1472,  829, 1473,  829,  764, 1474,  829, 1475,   81, 1476,
         1477, 1062,  102, 1063,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  246,  353,   19,  151,  233,  157,   62,  148, 1000,  157,   62,
          604,  242,   83,   16, 1001,   18,  172,  242, 1002, 1003,   10,  155,
         1004,   10,  154,  246,  353,   19,  151,  247,  366,  157,   62,  158,
          179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  13,   79,  159,   11,   12,  160,  161,   16,  162,   18,   17,   13,
           34,   79,   19,   17,  163,   23,  164,   13,   19,   79,   19,   29,
          589,   23,  164,   79,   34,   29,  589,   23,  166,  161,   24,  167,
           33,   10,   13,   34,  133,   79,  168,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  750,   63, 1647,    9,   92,   16,   29,   18,   28,   92,   16,
           29,   18,   65,   92,   16,   29,   18,   66,   92,   16,   29,   18,
           59,   63,   52,   68,  256,  558,  468,   92,   16,   29,   18,  814,
          172,  821,   92,   16,   29,   18,  241,  814,  815,  242,  472,   92,
           16,   29,   18,   65,   92,   16,   29,   18,  603,   10,    9,   92,
           16,   29,   18,  256,  153, 1600,   47,  256,   19,  462,   92,   16,
           29,   18,   65,   92,   16,   29,   18,   65,  247,    3,    3,    3,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 889,   16,  125,   18,   10,   91,   29,   26,   19,   29,   93,   26,
          889,   16,  170,   18,   10,   91,   19,   29,   26,   17,   93,   91,
           17,  889,   16,  125,   18,   34,  889,   16,  170,   18,   93,  263,
          889,   16,  125,   18,   10,   91,   27,   93,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   13, 1112,   15, 1762,  125,  248,   15,   16,   13,
           13,  250,  113,   13,  251,  252,  113,   13,  138,  253,  229,  125,
          254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 173,   79,   10,   17,  174,   13,   68,  579,  176,  178,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  112,  113,   91,   13,   93,   10,   36,   17,   13,   19,  125,
           36,   34,  125,  125,   10,   17,  249,  229,  290,  113,   91,   13,
           93, 1271,  105, 1272,  112,  477,   91,   13,   93,   10,   36,   17,
           13,   19,   29,   36,   13,  294,  573,  249,  113,   91,   13,   93,
           34,  477,   91,   13,   93,  165,   20,  229,  125,  254,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 180,   11,   12,  125,   92,   16,  131,   18,   24,   25,  102,  125,
           92,   16,   29,   18,   10,   17,   26,   27,  125,   92,   16,  131,
           34,   29,   18,   10,   17,  125,   92,   16,  131,   18,  353,   92,
           16,  131,   18,   11,   12,  125,   92,   16,  131,   18,   24,   25,
          181,  131,  183,  353,   92,   16,  131,   18,   10,  806,  131,   10,
          179,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7, 1158,  169,  125,  170,   36,  125,   36,   10,   17,   36,  170,
           36,   91,  125,   19,  170,   93,  172,  170,  125,  170,  350,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,   66,  232,  233,   62,  232,   59,  234,  603,   59,  604,  603,
          594,  145,  151,   59,  235,  236,  142,  237,  238,  605,   10,  116,
           16,  228,   16,  105,   18,   18,   16,  105,   18,  237,  241,   27,
          266,  172,  242,  606,  229,  607,   28,   19,  608,   19,  256,  609,
          610,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 980, 1682, 1683, 1684, 1685, 1683, 1684,  548, 1685,   44, 1686,  750,
          882, 1687, 1688, 1125, 1689, 1690, 1691, 1685,   44, 1383, 1692,  121,
         1383, 1692,  894,  894, 1693,  858,  688, 1383, 1692, 1694,  766,  736,
         1695,  121, 1696, 1697, 1383, 1692,  233, 1284,  882, 1069,  882, 1698,
         1383, 1692,  682, 1699, 1700, 1701, 1699, 1702,  670, 1703,  417, 1703,
          417,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 113,   91,   13,   93,   10,  114,   13,   19,  230,   13,  118,   23,
          584,  125,  119, 1627,  112,  125,  168,   91,   27,   93,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  915,   63,  102,  145,   83,   16,  151,   18,  236,  145,   83,
           16,  823,   18,  395,  239, 2016,   10,  239, 2017,  164,    9,   28,
           28,   66,  164,  241,  242,  823,  172,  242,  151,  155,  242,  361,
          813,  853,  256, 2018,  816,  915,   63,  407,  247,  220,  229,  607,
           66,  603,   65,  609,  610,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 353,   92,   16,  131,   18,  370,   11,   12,  125,   92,   16,  131,
           18,   24,   25,  181,  131,  183,  125,   92,   16,   29,   18,   10,
           29,  353,   92,   16,  133,   18,   10,  881,  301,  170,   92,   16,
          131,   18,   10,   11,  118, 1532,  125,   92,   16,  131,   18,   24,
          119,  118,   13,  119,  339,   13, 1546,  118, 1436,  119,   10,   23,
          118, 1532, 1622,  119,   10,   29,  229,  170,   92,   16,   29,   18,
          170,   92,   16,  482,   18,  170,   92,   16, 2086,   18,  229,  180,
           11,   12,  170,   92,   16,  131,   18,   24,   25,  181,  993,  183,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  230,   15,   16,   17,   18,   13,
           34,  228,   16,   20,   18,  114,   13,   19,  116,   16,   20,   18,
           16,   21,   18,   11,   91,   13,  294,   11,  118,   23,   26,  116,
           16,  117,   18,   16,   17,   18,   24,  119,   24,   93,  168,    4,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1011,  406,  924,  175, 1012,  893, 1013, 1014, 1015,  596, 1016, 1012,
          530,  414,  924, 1017, 1018,   81,  726, 1019, 1014,  300,  414,  182,
          924, 1018,  423,  425,   16,   13,   18,   10,  116,   16,   29,   18,
           16,  182,   18,  424,   92,   16,   35,   10,   29,   18,   15,   16,
          182,   18,   13,   92,   16,   35,   18,   10, 1020,   26,   27, 1021,
           10,  228,   16,  116,   16,   29,   18,   16,  182,   18,  424,   92,
           16,   35,   10,   29,   18,   15,   16,  182,   18,   11,   91,   13,
           92,   16,   35,   18,   19,  425,   16,   13,   18,   24,   93,   15,
           16,   17,   18,   18,   10,  228,   16,  116,   16,   29,   18,   16,
          182,   18,   11,   91,  424,   92,   16,   35,   10,   29,   18,   15,
           16,  182,   18,   13,   92,   16,   35,    0],
        [1601, 1602, 1603,  338, 1604, 1605, 1606,  413, 1607, 1608,  249,  530,
          414,  216, 1605, 1017, 1609, 1603, 1607,   81,  898,  942, 1101, 1103,
          231,   61,  894, 1606,  413, 1611, 1607,  880, 1734,  301,    9, 1455,
         1604, 1606, 1610, 1611, 1607, 1301, 1612,  338, 1606, 1610, 1611, 1607,
         1301, 1612,  219,    9,   71, 1613,  553,  553,  554, 1382,  556, 1611,
         1607, 1606,  413, 1611, 1607,  942, 1101, 1103, 2127, 1606, 2128, 1607,
         1379,  430,  435,  436,   83,   16,  503,   18,   15,   16,   17,   18,
           10,  116,   16,   83,   16,  131,   18,   91,   83,   16,  539,   18,
           19,   83,   16,  540,   18,   93,   15,   16,   17,   18,   18,   16,
           91,   83,   16,  125,   18,   34,   83,   16,  170,   18,   93,   91,
           83,   16,  162,   18,   34,   83,   16,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  255,  256,   64,  140,  141,  457,  116,   16,   29,   18,   16,
           17,   18,  256,  259,  139,  513,   65,   79,   15,   16,   17,   18,
           10,  134,   13,  139,  455,    9,   28,   65,  458,  256,  138,  267,
           36,  153,   36,   10,   91,   27,   93,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 882,   83,   16,   63,   18,   19,   83,   16,    9,   18,   92,   16,
           29,   18,   83,   16,   28,   18,   92,   16,   29,   18,   83,   16,
           65,   18,   92,   16,   29,   18,   83,   16,   66,   18,   92,   16,
           29,   18,  102,  256,  558, 1624,   92,   16,   29,   18,  466, 1625,
          107,  603,  269,  719, 1626,  741,   91,   27,   93,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 116,   16,   29,   34,   17,   35,   18,   16,   29,   19,   17,   35,
           18,   10,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 650,  651,  652,  653,   40,  654,  655,  656,  102,  657,  658,   64,
          659,  660,  661,  662,  663,  664,  665,  663,  666,  667,  668,  669,
          661,  670,  671,  672,  673,  300,  324,  667,  674,  670,  675,  674,
          676,  667,  672,  673,  300,  324,  667,    7,  677,  678,  679,  680,
          681,  667,  657,  682,  683,  684,  661,  685,  663,  664,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  112,  113,   91,   13,   93,   10,   83,   16,  248,   18,   15,
           16,   13,   18,   34,  641,   15,   16,   17,   18,   19,   13,  125,
           10,   29,  249,  250,  113,   91,   13,   93,  251,  252,   13,  165,
           23,  249,  113,   91,   13,   93,  165,  116,   16,   29,   18,   16,
           17,   18,   13,   15,   16,   20,   18,   34,   29,   26,  229,  125,
          254,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  239,  856,   10,  201,   15,   16,   82,   18,  142,  242,  151,
         1163,  238,  608,   10,   17,   68,  142,  239,  856, 1164,  155,  584,
          154,  358, 1165,  228,   16,   20,   18,  142,  242,  151,  358,    4,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,   52,   63,  979,  980,   40,  981,   52,  982,  102,  983,  984,
           52,   64,  985,  986,   52,  530,  987,  238,  988,  989,  983,   71,
            3,    3,    3,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 127,  129,  355,   13,   10,   23,   26,   27,   79,   10,   29,   26,
           27,  131,   10,   29,  130,   13,   79,  610,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  125,   10,  126,   92,   16,   17,   18,  841,   26,  170,   10,
           17,   15,   16,  841,   18,   26,  162,   10,  841,   15,   16,  901,
           18,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1507,   65,  137,  116,   16,   13,   15,   16,   17,   18,   18,   16,
          125,   15,   16,   17,   18,   18,   34,  116,   16,   79,   15,   16,
           17,   18,   18,   16,  170,   15,   16,   17,   18,   18,   10,   29,
           26,   27,   91,  125,  257,  170,  257,   23,   93,  457,  116,   16,
          228,   16,   17,   18,   18,   16,   17,   18,   26,   68,  228,   16,
           17,   18,   65, 1508,  255,   65,  178,  107,  277,  141,   62,  382,
          356,  277,   65,  138,  267,    9,   28, 1089,  153,  466,  464,  241,
          107,  848,  177,  277,  177, 1509, 1185,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [2087, 1549,  125,   81,  202,  918, 2087, 1550, 1551, 1551, 1552, 1549,
          825, 1553, 1554, 1555, 2088, 1548, 1551, 1050, 1553, 1554,   71, 2089,
         1551, 1552, 1549, 1557, 1549,   71, 1551, 1552, 1549, 1557, 1549, 2090,
         1549, 1549, 2091, 2092,   71,  229, 1551, 1552,  544, 1549, 1549,   60,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  232,   59,  523,  524,  525,  232,  523,  157,  247,  179,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 800,   50,   59,  713,  182,  520,  231,  800, 1069,   68,  464,  691,
         1070,    9,  800, 1069,   68,  465,  258, 1071, 1070,   28,  800,   55,
          464,  465, 1072,  102, 1073, 1072,  110,    3,    3,    3,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,  643,   10,   12,   19,   17,   26,   19,   29,   26,   23,
           26,   29,   26,   17,   26,   20,   25,    9,   10,   12,   19,   29,
           26,   23,   26,   29,   25,   28,   10,   12,   29,   26,   17,   25,
           65,   92,   16,  643,   18,   91,    9,  644,   28,   93,   10,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 363,  151,  226,    9,   28,   65,  125,  170,  162,  114,    9,   10,
          116,   16,   21,   18,   16,   30,   18,  114,   65,   10,  116,   16,
           30,   18,   16,  729,   18,  125,  170,  179,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,    8,    9,   10,   12,   23,   26,   17,   26,   21,   26,  105,
           26,  134,   26,   96,   25,   26,   28,   10,   12,   21,   26,  134,
           25,   65,   92,   16,    9,   18,   28,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,    9,   26,   28,   26,   65,  147,   62,  148,  149,   68,  150,
           62,   92,   16,   29,   18,  145,  151,  152,  150,   62,   92,   16,
           29,   18,   53,   21,  117,  153,   10,  154,   10,  155,   10,  156,
           92,   16,   29,   18,   26,  157,   62,  158,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [1361,  441,  893, 1362, 1363,  710, 1364, 1365,  893,  894,  338, 1366,
          894, 1366, 1362, 1367,   97,  705, 1368,  530,  407, 1369, 1370,   85,
         1371, 1368, 1372, 1366, 1373, 1368, 1374, 1366, 1368,  893,  331, 1375,
           81, 1016, 1193, 1376, 1377, 1376, 1377,  554, 1366, 1362, 1378,  449,
          229,   97,  705, 1368,  893,  828, 1375, 1379,  171,  893, 1380, 1381,
          339,  171,  339,  171, 1368,  433,  997,  553,  102,  553, 1382,  556,
          894, 1366, 1362,  440,  450,   83,   16,  503,   18,   15,   16,   17,
           18,   10,  116,   16,   83,   16,  131,   18,   91,   83,   16,  539,
           18,   19,   83,   16,  540,   18,   93,   15,   16,   17,   18,   18,
           16,   91,   83,   16,  125,   18,   34,   83,   16,  170,   18,   93,
           91,   83,   16,  162,   18,   34,   83,    0],
        [ 384,   26, 1750,  138,  242,  171,  131,  359,  107,   99,  372,  171,
          172,  131,  171,  172,  384,  131,  717, 1750,  384,  172, 1750,  171,
          172,  384,  131,  717,  384,  171,  172,  131,  384,  717, 1750,  171,
          383,  384,  171,  717, 1750,  171, 2084,  384,  717, 1750,  171,  384,
         1074,  647,  131, 1750, 1074,  647, 1063,  372,  179, 2085,  387,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,  113,   91,   13,   93,   10,  114,   91,  115,   13,   34, 1738,
           93,  797,  798,  113,   91,   13,   93,  251,  799,  546,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  125,  294,   11,   91,   23,   26,  116,   16,  117,   18,   16,
           17,   18,   24,   93, 1665, 1393, 1666, 1393,  886, 1393,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  113,   91,   13,   93, 1066,   13,   22,   23,  249,  113,   91,
           13,   93,   10,   19,  248,   15,   16,  641,   18,  113,   91,  174,
           17,   93,   10,  134,    4,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  557,   63,  153,  154,   62,  153,  466,   68,  142,  154,  269,
         1088, 1182,  301,  239, 1485,   10,   13, 1486,  142,    9,   28,  362,
          358,   13,  112,  113,   91,   13,   93,   79,   10,  113,   91,   13,
           93,  797,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  125,   26,  170,  349,   36,  125,   34,  170,   36,   10,   29,
           26,   36,  125,   19,  170,   36,   10,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[  61,  112,  113,   91,   13,   93,   10,  114,   11,   91,  115,   13,
           34,  116,   16,  117,   18,   16,  105,   18,   24,   93,  118,   19,
          117,   26,  117,  119,  120,  121,  113,   91,   13,   93,  122,  123,
            3,    3,    3,    3,    3,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   83,   16,   79,   18,   10,   83,   16,    9,   18,  230,   91,
          115,   83,   16,   13,   18,   34,  796,   93,  797,  798,    3,    3,
            3,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,  230,   11,   91,  115,   13,
           34,  116,   16,  117,   18,   16,   30,   18,   24,   93,   91,  115,
          257,   23,   93,    7,  113,   91,   13,   93,  118,   23,   26,   17,
          117,  119,  380,  381,  253,  385,   99, 1062,  113,   91,   13,   93,
           91,   23,   26,   17,  117,   93,  380,  381, 1964,   68,  113,   91,
           13,   93,   91,   23,   26,   17,  117,   93,  380,  381, 2032,   68,
          113,   91,   13,   93,   11,   91,   23,   26,  116,   16,  117,   18,
           16,   96,   18,   24,   93,  251,  888,  254,   11,  118,  116,   16,
          298,   18,   16,   30,   18,   26,  116,   16, 2033,   18,   16,   96,
           18,   24,   93, 1063, 1062, 1148,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 132,   11,   12,  125,   92,   16,  131,   18,   24,   25,  102,   26,
           27,  125,   92,   16,   29,   18,   10,   29,   26,   27,  125,   92,
           16,   30,   18,   10,   21,  125,   92,   16,   20,   18,  229,   11,
           12,  125,   92,   16,  131,   18,   24,   25,  526,  527,  301, 1424,
           11,   12,  125,   92,   16,  131,   18,   24,   25,  181,  131,  183,
          353,   92,   16,  171,   18,   10, 1360,  229,  171,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[274, 275, 276, 102, 107, 277, 137,  27,  79,  10, 279,  27,  91, 279,
         572,  23,  93, 361,  79, 949,  68, 464, 361, 513,  65, 137,  27,  79,
          15,  16,  17,  18,  10,  17, 514,  91, 373, 257,  23,  93, 853, 142,
         584, 464,  68, 142, 950, 465, 951, 849, 911, 361,  65, 853, 952, 839,
         229, 116,  16,  36, 953,  36,  18,  16,  36, 849,  36,  18, 842, 954,
         952, 107, 955,  65, 289,  68, 449,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [738,   7, 112, 113,  91,  13,  93,  10,  17,  13,  15,  16,  20,  18,
          19, 641,  15,  16,  17,  18,  34,  17, 250, 113,  91,  13,  93, 251,
         252,  23,  22, 125,  22,  20, 249, 301, 113,  91,  13,  93, 546, 168,
         464, 111, 171, 229, 464,  19, 171, 254,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 61,  62, 140, 141, 107,  13,  10,  17, 513,  79,  15,  16,  17,  18,
          10,  17, 514,  91, 373, 257,  23,  93, 515,  66, 256, 362, 516, 172,
         517,  65, 139, 140,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0],
        [ 13,  79, 159,  11,  12, 160, 161,  16, 277,  18,  13,  19,  79,  34,
          29, 589,  23, 164,  13,  19,  17,  79, 163,  23, 164,  13,  34,  17,
          79,  19,  17, 163,  23, 166, 161,  24, 167,  33,  13,  79, 168, 737,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,   0,
           0,   0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[   7,  178,  116,   16,   13,   15,   16,   17,   18,   18,   16,  171,
           15,   16,   17,   18,   34,  131,   18,   19,  116,   16,   79,   15,
           16,   17,   18,   18,   16,   20,  171,   15,   16,   17,   18,   19,
          131,   18,   10,   29,  136,  136,  220,  139, 1381,  358,  131,  254,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  169,  889,   16,  125,   18,   10,   91,   29,   26,   17,   93,
           26,  889,   16,  170,   18,   10,   91,   17,   26,   19,   17,   93,
           26,  889,   16,  162,   18,   10,   91,   29,   26, 1348,   93,  167,
          261,   16,   83,   16,  162,   18,   18,  469,  469,   91,   17,  261,
           16,   83,   16,  125,   18,   18,   34,  261,   16,   83,   16,  170,
           18,   18,   93, 1348,   10,    4,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 301,  353,   92,   16,  131,   18,  370,   11,   12,  125,   92,   16,
          131,   18,   24,   25,  181,  131,  183,    7,  353,   92,   16,   21,
           18,   10,   23,  584,  125,   92,   16,   30,   18,   10,   30,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  145,  956,  957,  669,  239,   83,   16,  958,   18,   10,  819,
           15,   16,   82,   18,   62,  234,  854,  523,  959,  241,  107,  153,
          150,   62,  357,   68,   65,  584,   66,  150,   62,    9,  584,   28,
          584,   65,  584,   66,  960,  961,  241,   83,   16,  153,   18,  469,
          469,   83,   16,  269,   18,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 250,  112,  113,   91,   13,   93,   10,  116,   16,   13,   19,   17,
           18,   16,   13,   34,   17,   18,  248,   15,   16,   13,   18,  251,
          252,  241,   13,  962,   23,  249,   91,   13,   19,   17,   93,  248,
           15,   16,   13,   18,   34,   13,   34,   17,  257,   23,  241,  125,
          294,  118,   23,   26,   29,   93,  249,  112,  477,   91,   13,   93,
           10,  116,   16,  248,   15,   16,   13,   18,   19,  641,   19,  125,
           18,   16,   13,   15,   16,   17,   18,   18,   27,   91,   13,  257,
           23,   93,  111,   61,  477,   91,   13,   93,  111, 1758,   91,  125,
           93, 2093, 1758,   91,  125,   93, 1531,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [  61,  112,  113,   91,   13,   93,   10,   13,   15,   16,   20,   18,
           19,  116,   16,   29,   18,   16,   13,   15,   16,   20,   18,   18,
           26,  113,   91,   13,   93,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,  125,  170,  349,  125,  170,  162,   10,   17,  125,   19,  228,
           16,   30,   18,  170,  114,   22,  889,   16,  125,   18,   26,  889,
           16,  162,   18,  257,   10, 2034,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [   7,    8,    9,   10,   12,   13,   36,   13,   22,   17,   25,   28,
           10,   12,   13,   36,   20,   19,   17,   13,  257,   23,   25,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 231,   68,  465,   52,   63,   64,  363, 1413,  236,  242, 1413,  172,
          242,   63,   26,  464, 1089,  561,  466,    4,    3,    3,    3,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 145,  151,  226,    9,   28,   65, 2184,  125,  170,  162,  170,   10,
          105,   26,  125,   10,   17,  162,   26,   28,   10,  116,   16,  117,
           18,   16,   20,   18,  145,  151,   53,  179,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  764, 1224,  829, 1225,  829, 1226,  707, 1227, 1228,   81, 1229,
         1230, 1231, 1232, 1233,  822, 1232, 1224,  822, 1225, 1234, 1235,   90,
          164, 1236,   16,   35,   15,   16,   79,   16,   35, 1237,  164, 1236,
           16,   35,   15,   16,  279,   16,   35,   79,   16,   35, 1238,  164,
         1239,   16,  164, 1236,   16,   35,   15,   16,  164, 1240,   79,   16,
           35,  164, 1241,   16,   79,  164,   15,   16, 1242,  164, 1239,   16,
          164, 1243, 1244, 1235,  527,  434,  451,   10,  116,   16,  424,   92,
           16,   35,   10,   29,   18,   15,   16,  131,   18,   11,   91,  279,
           92,   16,   35,   18,   19,  425,   16,  279,   18,   24,   93,   11,
           91,   79,   92,   16,   35,   18,   19,  425,   16,   79,   18,   24,
           93,   18,   16,  228,   16,  424,   92,    0],
        [ 518,  519,  520,  519,  158,    3,    3,    3,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _th_index_select
[DEBUG]Sample idx: tensor([[ 274,  275,  276,  102,  173,   65,  273,  178,   11,   12,  160,  161,
           16,  277,   18,   13,   10,   17,  114,  288,  164,   79,   10,   21,
          230,  288,  166,  161,   24,  167,  288,  164,  273,  107,  277,  273,
          178,   11,   12,  160,  161,   16,  277,   18,   13,   10,   29,   34,
          279,  114,  384,  164,   79,   10,   17,   34,  279,  230,  384,  166,
          161,   24,  167,  279,  273,  229,   65,  277,  274,  178,  173,   65,
         1646,  107,  277,  877, 1089,  466,  140,  229,  277,  177,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 521,  125,   10,  116,   16,   20,   18,   16,   21,   18,  114,   15,
           16,   17,   18,  125,   34,   17,  230,   17,  125,   10,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 231,  153,  852,   62,  604,  155,  852,   62,  176,  154,  361,  852,
           62,  853,  256,   66,  155,  466,  241,  560,  852,   62,  176,   83,
           16,  854,   18,   10,  228,   16,   20,   18,   83,   16,  855,   18,
          229,  818,  856,  857,    3,    3,    3,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0],
        [ 112,   79,   10,   29,   34,   13,   34,  116,   16,  230,   13,   18,
           16,   13,   15,   16,   17,   18,   18,  797,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,    0,
            0,    0,    0,    0,    0,    0,    0,    0]], device='cuda:0')
Train time:  5.155178070068359

'../../examples/test_model/elmo'

3.使用模型

3.1使用训练好的Elmo模型

item = [
        {'stem': '如图$\\FigureID{088f15ea-8b7c-11eb-897e-b46bfc50aa29}$, \
        若$x,y$满足约束条件$\\SIFSep$，则$z=x+7 y$的最大值为$\\SIFBlank$'},
        {'stem': '已知圆$x^{2}+y^{2}-6 x=0$，过点(1,2)的直线被该圆所截得的弦的长度的最小值为'}
]

tokenizer_kwargs = {"path": os.path.join(output_dir, "vocab.json")}
i2v = Elmo('elmo', 'elmo', output_dir, tokenizer_kwargs=tokenizer_kwargs)

# 可以对单个题目进行表征
i_vec, t_vec = i2v(item[0]['stem'])
print(i_vec.shape) # == torch.Size([x])
print(t_vec.shape) # == torch.Size([x, x])

# 也可以对题目列表进行表征
i_vec, t_vec = i2v([ item[0]['stem'], item[1]['stem'] ])
print(i_vec.shape) # == torch.Size([2, x])
print(t_vec.shape) # == torch.Size([2, x, x]))

torch.Size([512])
torch.Size([15, 512])
torch.Size([2, 512])
torch.Size([2, 25, 512])

3.2使用Elmo Tokenizer

# 加载之前训练的模型tokenizer
tokenizer = ElmoTokenizer(os.path.join(output_dir, "vocab.json"))

# 对题目文本进行令牌化
items = [
    "有公式$\\FormFigureID{wrong1?}$，如图$\\FigureID{088f15ea-xxx}$,\
    若$x,y$满足约束条件公式$\\FormFigureBase64{wrong2?}$,$\\SIFSep$，则$z=x+7 y$的最大值为$\\SIFBlank$",
    "已知圆$x^{2}+y^{2}-6 x=0$，过点(1,2)的直线被该圆所截得的弦的长度的最小值为"
]
# 可以对单个题目进行令牌化
print(tokenizer(items[0], freeze_vocab=True))
print()

# 也可以对题目列表进行令牌化
print(tokenizer(items, freeze_vocab=True))
print()

([527, 231, 3, 13, 26, 79, 159, 527, 6, 33, 10, 13, 34, 133, 79, 168, 4], 17)

([[527, 231, 3, 13, 26, 79, 159, 527, 6, 33, 10, 13, 34, 133, 79, 168, 4], [7, 104, 13, 15, 16, 17, 18, 34, 79, 15, 16, 17, 18, 19, 105, 13, 10, 23, 106, 107, 104, 108, 109, 110, 111]], [17, 25])

# 可以使用tokenize方法查看令牌化后的文本
print(tokenizer.tokenize(items[0], freeze_vocab=True))
print(tokenizer.tokenize(items, freeze_vocab=True))

['公式', '如图', '[FIGURE]', 'x', ',', 'y', '约束条件', '公式', '[SEP]', 'z', '=', 'x', '+', '7', 'y', '最大值', '[MARK]']
[['公式', '如图', '[FIGURE]', 'x', ',', 'y', '约束条件', '公式', '[SEP]', 'z', '=', 'x', '+', '7', 'y', '最大值', '[MARK]'], ['已知', '圆', 'x', '^', '{', '2', '}', '+', 'y', '^', '{', '2', '}', '-', '6', 'x', '=', '0', '过点', '直线', '圆', '截得', '弦', '长度', '最小值']]

3.3使用EduNLP中公开的预训练模型

# 获取公开的预训练模型
pretrained_dir = f"{BASE_DIR}/examples/test_model/elmo"
i2v = get_pretrained_i2v("elmo_test", model_dir=pretrained_dir)

EduNLP, INFO model_path: ..\..\examples\test_model\elmo\elmo_test
EduNLP, INFO Use pretrained t2v model elmo_test
downloader, INFO http://base.ustc.edu.cn/data/model_zoo/modelhub/elmo_pub/1/elmo_test.zip is saved as ..\..\examples\test_model\elmo\elmo_test.zip

Downloading ..\..\examples\test_model\elmo\elmo_test.zip 100.00%: 402KB | 402KB

downloader, INFO ..\..\examples\test_model\elmo\elmo_test.zip is unzip to ..\..\examples\test_model\elmo\elmo_test

items = [
    "有公式$\\FormFigureID{wrong1?}$，如图$\\FigureID{088f15ea-xxx}$,\
    若$x,y$满足约束条件公式$\\FormFigureBase64{wrong2?}$,$\\SIFSep$，则$z=x+7 y$的最大值为$\\SIFBlank$",
    "已知圆$x^{2}+y^{2}-6 x=0$，过点(1,2)的直线被该圆所截得的弦的长度的最小值为"
]
i_vec, t_vec = i2v(items)
print(i_vec.shape)
print(t_vec.shape)
print()

# 也可以单独获取题目表征和各个token的表征
i_vec = i2v.infer_item_vector(items)
print(i_vec.shape)
t_vec = i2v.infer_token_vector(items)
print(t_vec.shape)
print()

# 同样，可以获取单个题目的表征
i_vec, t_vec = i2v(items[0])
print(i_vec.shape)
print(t_vec.shape)

torch.Size([2, 64])
torch.Size([2, 25, 64])

torch.Size([2, 64])
torch.Size([2, 25, 64])

torch.Size([64])
torch.Size([17, 64])