You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
shenyan ea7bd5d762 修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方 4 years ago
free_servers 完成神经网络训练框架; 完成分类任务 4 years ago
network 修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方 4 years ago
network_module 完成神经网络训练框架; 完成分类任务 4 years ago
custom_profile.py 完成神经网络训练框架; 完成分类任务 4 years ago
data_module.py 修改了evaluate中的拼写错误;使用自定义激活函数jdlu; 添加由版本名获得对应版本的ckpt的函数;改变存在保存路径的情况下的保存模式; 开发k折交叉验证; 增加了excel表格读写函数 4 years ago
evaluate.py 修改了evaluate中的拼写错误;使用自定义激活函数jdlu; 添加由版本名获得对应版本的ckpt的函数;改变存在保存路径的情况下的保存模式; 开发k折交叉验证; 增加了excel表格读写函数 4 years ago
main.py 修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方 4 years ago
requirements.txt 完成神经网络训练框架; 完成分类任务 4 years ago
save_checkpoint.py 修改了evaluate中的拼写错误;使用自定义激活函数jdlu; 添加由版本名获得对应版本的ckpt的函数;改变存在保存路径的情况下的保存模式; 开发k折交叉验证; 增加了excel表格读写函数 4 years ago
train_model.py 修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方 4 years ago
utils.py 修改根据版本号和当前fold数得到ckpt地址的函数的BUG; 加入基于JDLU激活函数的网络; 改变网络的初始化方 4 years ago

基于pytorch lightning的机器学习模板, 用于对机器学习算法进行训练, 验证, 测试等, 目前实现了神经网路, 深度学习, k折交叉, 自动保存训练信息等.

Python Text Shell

Contributors (1)