shenyan
/
machineLearningTemplate
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
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.
3 Commits
1 Branch
54 kB
9 Download
 
 
 
Tree: b114979ea5
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b114979ea5'
${ noResults }
shenyan b114979ea5 改用pin_memory=True;添加对gpu性能的注意说明;自动获取核心数并设置其为num_workers;简化每个epoch的随机数更换逻辑 4 years ago
free_servers 完成神经网络训练框架; 完成分类任务 4 years ago
network 将模型改为拟合函数; 可视化拟合函数的结果 4 years ago
network_module 完成神经网络训练框架; 完成分类任务 4 years ago
custom_profile.py 完成神经网络训练框架; 完成分类任务 4 years ago
data_module.py 改用pin_memory=True;添加对gpu性能的注意说明;自动获取核心数并设置其为num_workers;简化每个epoch的随机数更换逻辑 4 years ago
evalute.py 完成神经网络训练框架; 完成分类任务 4 years ago
main.py 改用pin_memory=True;添加对gpu性能的注意说明;自动获取核心数并设置其为num_workers;简化每个epoch的随机数更换逻辑 4 years ago
requirements.txt 完成神经网络训练框架; 完成分类任务 4 years ago
save_checkpoint.py 改用pin_memory=True;添加对gpu性能的注意说明;自动获取核心数并设置其为num_workers;简化每个epoch的随机数更换逻辑 4 years ago
train_model.py 将模型改为拟合函数; 可视化拟合函数的结果 4 years ago
utils.py 完成神经网络训练框架; 完成分类任务 4 years ago
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

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

Python Text Shell

Contributors (1)
All