Skip to content
@tmu-sugimura-lab

杉村研究室

杉村研究室

研究活動を支えるコード・ドキュメントをまとめています.


📘 Handbook

環境構築をはじめ,日々の研究に役立つ Tips をまとめています.

handbook


🚀 train4all

PyTorch 向けのミニマルな機械学習フレームワークです. BaseTrainer を継承し,setup()compute_loss()compute_metrics() の3つを実装するだけで,チェックポイント保存,early stopping,メトリクスの記録・可視化,ログ出力,ライブ Web ダッシュボードまでを自動でこなします. 混合精度学習や勾配累積による省メモリ化にも対応し,小規模な GPU でも扱えます.

train4all


🤝 Contribution

研究室内で役立つ情報や改善案があれば,Issue や Pull Request で気軽に共有してください.

Popular repositories Loading

  1. .github .github Public

    1

Repositories

Showing 1 of 1 repositories

Top languages

Loading…

Most used topics

Loading…