pytorch_toolbox pytorch的有用工具库 focus_loss_multiclass.py 焦损类为多类。可以将Γ,权重(对应于原始纸张的alpha),size_average设置为参数。 numpy_loader.py 用于保存为npy文件的数据集的数据加载器。
评论0
最新资源