[DEV] python的nn.Module中添加 to函数 #890

Open

opened

目标版本
main
功能描述

python的nn.Module中添加 to函数：
实现将model权重to到gpu上的操作
model.to("cuda")

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests