一些ai开发过程中使用到的工具模块
Project description
aikit
一些ai开发过程中使用到的工具模块
安装
git clone git@github.com:cpcgskill/aikit.git
使用
aos
import os
# 设置以下环境变量为腾讯云cos的配置
os.environ['COS_Region'] = 'ap-hongkong'
os.environ['COS_SecretId'] = ''
os.environ['COS_SecretKey'] = ' '
os.environ['COS_Bucket'] = ' '
import torch
from aidevkit.aos import Saver
class MyModule(torch.nn.Module):
def __init__(self):
super(MyModule, self).__init__()
self.linear = torch.nn.Linear(10, 10)
self.s = 1
def forward(self, x):
return self.linear(x)
saver = Saver(lambda: MyModule(), 'test.pt', 3)
for i in range(10):
saver.step()
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
aidevkit-0.3.3.tar.gz
(9.5 kB
view hashes)
Built Distribution
Close
Hashes for aidevkit-0.3.3-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 894e5efba4823d6eeb1f4e56f251a4b90255c4613e9fc8b53b15e161173c7d48 |
|
MD5 | c1e15838c26c27008205d6cc64c4afe5 |
|
BLAKE2b-256 | 7df003e9ca0cbf5ec8e5dd6503157a1af3d4f9f249d9e0863ae456816b4c6770 |