一些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.2.0.tar.gz
(9.3 kB
view hashes)
Built Distribution
Close
Hashes for aidevkit-0.2.0-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5697a567b6d2690273a33798d85abfbf203b0b265132a0cee27dd109735beb59 |
|
MD5 | 32254444eb2ca6f0ae1cc0c9f14dd25c |
|
BLAKE2b-256 | cafd8ca02ac8985d92ee21b42a7bfc5c19bd08cd83ac3433218383b2a0eb6fa6 |