一些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.1.tar.gz
(9.5 kB
view hashes)
Built Distribution
Close
Hashes for aidevkit-0.3.1-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1f8b80e387553c776d8d5ac6f423085184f1adc14312d3c70bfd64511b862345 |
|
MD5 | de0d5126a91513995c5c93e1f1cfbbca |
|
BLAKE2b-256 | 9262f1f430eafa8c80a62e2a44a6dbc92929f13c6b2c4047bfe8631f086a9aef |