一些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.0.tar.gz
(9.5 kB
view hashes)
Built Distribution
Close
Hashes for aidevkit-0.3.0-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 29b5c6f0ba24795dc0e787c8d35c0229aa22dd401bc02f844c8356fb1e74a6b4 |
|
MD5 | 50339be975862eacdd94eafd363090c9 |
|
BLAKE2b-256 | 5c4c0666a57be0c497b58e7165bcd4593ffc95fbd990a83e8c143bf8bbd200cd |