implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.99k
stars
518
forks
source link
FIX: RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) and RuntimeError: result type Float can't be cast to the desired output type long int #284
FIX: RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) and RuntimeError: result type Float can't be cast to the desired output type long int
FIX: RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) and RuntimeError: result type Float can't be cast to the desired output type long int