Parameter(torch.from_numpy(gen_emb), requires_grad=False) ... batch_test_X_mask=(test_X[offset:offset+batch_size]!=0)[batch_idx].astype(np.uint8) ...
確定! 回上一頁