加载SST的label是从1开始的
而CrossEntropyLoss这个函数要求label是从0开始的
否则种数就会不匹配
报错:
RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed.
加载SST的label是从1开始的
而CrossEntropyLoss这个函数要求label是从0开始的
否则种数就会不匹配
报错:
RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed.