Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-25 05:22:43 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
65ee6dcc205abb703e09b0ef3ef6d3571d87cd14
ColossalAI/colossalai/nn
History
Ziyue Jiang 0653c63eaa [Tensor] 1d row embedding (#1075)
* Add CPU 1d row embedding

* polish
2022-06-08 12:04:59 +08:00
..
_ops
reorgnize colotensor directory (#1062)
2022-06-03 18:04:22 +08:00
layer
[Tensor] 1d row embedding (#1075)
2022-06-08 12:04:59 +08:00
loss
[hotfix] Raise messages for indivisible batch sizes with tensor parallelism (#622)
2022-04-02 16:12:04 +08:00
lr_scheduler
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
metric
[hotfix] Raise messages for indivisible batch sizes with tensor parallelism (#622)
2022-04-02 16:12:04 +08:00
model
Develop/experiments (#59)
2021-12-09 15:08:29 +08:00
optimizer
reorgnize colotensor directory (#1062)
2022-06-03 18:04:22 +08:00
parallel
[Tensor] fix optimizer for CPU parallel (#1069)
2022-06-06 17:36:11 +08:00
__init__.py
[refactory] add nn.parallel module (#1068)
2022-06-06 15:34:41 +08:00
init.py
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
Powered by Gitea Version: 1.25.2 Page: 1758ms Template: 65ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API