targets = data["targets"].to(self.device)
[[0, 1, 0, ... ,0]] # вложенный список на 11 позиций
targets = data["targets"].to(self.device)[0][:1] # просто первый элемент из 11 позиций возвращаю как срез...
input('Do you want to stop?')
stopFlag.changeFlag()