mixmatch-pytorch
使用PyTorch实现MixMatch( )
安装
pip install git+https://github.com/FelixAbrahamsson/mixmatch-pytorch
指示
该软件包提供了一个类mixmatch_pytorch.MixMatchLoader ,该类的工作方式与普通的PyTorch DataLoader一样,还提供了一个由mixmatch_pytorch.get_mixmatch_loss构造的损失函数。 使用示例,请参见下文。
您必须提供一个数据加载器,该数据加载器具有可重复使用的屈服字典,其键'features'和'targets'具有标记数据集的增强(!)功能和目标。 还必须为未标记的数据提供数据集,该数据集可以包装在PyTorch DataLoader中。 数据集必须返回带有'features'增强功能的
评论0
最新资源