However, you must use the with torch.jit.optimized_execution() context to trace or script your models ... pred_indices = torch.topk(probs, 5) pred_probs ...
確定! 回上一頁