I used the blocking launches before trying to train the model. Got the long error and then tried to print the labels but got this.
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-38-e847ac6dc72c> in <module>
----> 1 print(inputs['labels'])

C:\ProgramData\Anaconda3\lib\site-packages\torch\tensor.py in __repr__(self)
    177             return handle_torch_function(Tensor.__repr__, relevant_args, self)
    178         # All strings are unicode in Python 3.
--> 179         return torch._tensor_str._str(self)
    180 
    181     def backward(self, gradient=None, retain_graph=None, create_graph=False):

C:\ProgramData\Anaconda3\lib\site-packages\torch\_tensor_str.py in _str(self)
    370 def _str(self):
    371     with torch.no_grad():
--> 372         return _str_intern(self)

C:\ProgramData\Anaconda3\lib\site-packages\torch\_tensor_str.py in _str_intern(self)
    350                     tensor_str = _tensor_str(self.to_dense(), indent)
    351                 else:
--> 352                     tensor_str = _tensor_str(self, indent)
    353 
    354     if self.layout != torch.strided:

C:\ProgramData\Anaconda3\lib\site-packages\torch\_tensor_str.py in _tensor_str(self, indent)
    239         return _tensor_str_with_formatter(self, indent, summarize, real_formatter, imag_formatter)
    240     else:
--> 241         formatter = _Formatter(get_summarized_data(self) if summarize else self)
    242         return _tensor_str_with_formatter(self, indent, summarize, formatter)
    243 

C:\ProgramData\Anaconda3\lib\site-packages\torch\_tensor_str.py in __init__(self, tensor)
     83         if not self.floating_dtype:
     84             for value in tensor_view:
---> 85                 value_str = '{}'.format(value)
     86                 self.max_width = max(self.max_width, len(value_str))
     87 

C:\ProgramData\Anaconda3\lib\site-packages\torch\tensor.py in __format__(self, format_spec)
    532             return handle_torch_function(Tensor.__format__, relevant_args, self, format_spec)
    533         if self.dim() == 0:
--> 534             return self.item().__format__(format_spec)
    535         return object.__format__(self, format_spec)
    536 

RuntimeError: CUDA error: device-side assert triggered