上传者: 42108054
|
上传时间: 2021-12-30 20:59:51
|
文件大小: 13KB
|
文件类型: -
混合搭配
这是MixMatch的非官方PyTorch实现。 Tensorflow的官方实现在。
现在只有在CIFAR-10上的实验可用。
该存储库认真执行了官方实施的重要细节,以重现结果。
要求
Python 3.6+
PyTorch 1.0
torchvision 0.2.2(旧版本与此代码不兼容)
张量板
进步
matplotlib
麻木
用法
火车
通过CIFAR-10数据集的250个标记数据训练模型:
python train.py --gpu --n-labeled 250 --out cifar10@250
通过CIFAR-10数据集的4000个标记数据训练模型:
python train.py --gpu --n-labeled 4000 --out cifar10@4000
监控培训进度
tensorboard.sh --