TorchUtils:(WIP)TorchUtils是一个pytorch库,其中包含一些有用的工具和培训技巧

时间:2024-03-30 10:28:10
【文件属性】:

文件名称:TorchUtils:(WIP)TorchUtils是一个pytorch库,其中包含一些有用的工具和培训技巧

文件大小:40KB

文件格式:ZIP

更新时间:2024-03-30 10:28:10

Python

火炬工具 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


【文件预览】:
TorchUtils-master
----__init__.py(26B)
----torch_utils()
--------models()
--------tools.py(521B)
--------advanced()
--------dataset()
--------__init__.py(202B)
--------criterion()
--------layers.py(10KB)
--------__version__.py(62B)
--------optimizer()
--------lr_scheduler()
----README.md(4KB)

网友评论