tvietphu

This error occur because the function safe_to_tensor in imitation/src/imitation/util/util.py return a tensor without transferring to 'cuda'. fix the safe_to_tensor function: replace line 259: return array by the code: return th.as_tensor(array,...