先说torch版本:
torch 1.10.0 pypi_0 pypi
torchaudio 0.10.0 py39_cu113 pytorch
torchmetrics 0.6.2 pypi_0 pypi
torchvision 0.11.1 py39_cu113 pytorch
首先使用active_collection函数预处理数据,然后得到x_train_data, x_test_data, y_train_data, y_test_data用于训练和测试设计的分类器。
后面两行做了一个切片操作,就是y_train_data和y_test_data里面有多个label列,每次实验的时候我只选其中的一列。
with torch.no_grad(