火炬工具
TorchUtils是一个pytorch库,其中包含几个有用的工具以及一些最新的培训方法或技巧。 (工作正在进行中)
待办事项:整理代码库
使用pytorch 1.6重新编写仓库(因为PyTorch 1.6现在支持许多工具功能或技巧)
进口
import torch_utils as tu
全部播种
SEED = 42
tu.tools.seed_everything(SEED)
数据扩充
去做:
比赛中使用的通用数据扩充
模型
建议的预训练模型:
SEResNext-50
高效GPU
swsl_ResNeXt
EfficientNet_ns
混合网
网络
来自github仓库:
使用torch_utils快速构建模型:
import timm
model = timm.create_model('tresnet_m', pretrained=True)
model
2021-03-10 18:06:35
40KB
Python
1