library for deep learning and privacy preserving deep learning
Project description
hideandseek
privacy preserving deep learning library.
Why use hideandseek
?
-
Easy training & saving deep learning models along with other modules (ex: preprocessing modules) required in inference
-
Run multiple deep learning experiments in parallel on multiples GPUs (powered by hydra, and python multiprocessing)
-
Design and analyze experiments scientifically by modifying variables (powered by hydra)
-
Modularized machine learning pipeline allows using the same script for all types of experiments
-
The same training code can be run in privacy preserving setting by minimal modifications
Currently integrating from experiment codes. (30.10.2021.)
import torch
from omegaconf import OmegaConf
import hideandseek as hs
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
cfg = OmegaConf.load('config.yaml') # omegaconf.OmegaConf.DictConfig object
model = DNN() # torch.nn.Module object
train_dataset = dataset # torch.utils.data.Dataset object
kwargs = {
'model': model,
'dataset': train_dataset,
'cfg_train': cfg,
'criterion': criterion,
}
node = hs.Node(**kwargs)
node.model.to(device)
node.step(local_T=20, horizon='epoch') # trains for 20 epochs
# node.step(local_T=1000, horizon='step') # trains for 1000 steps
node.model.cpu()
node.save()
test_results = hs.eval.test(node)
scores = hs.eval.scores(test_results)
To do
- Migrate modules from experiment codes
- Draw figures to explain hideandseek
- GUI for generating experiment scripts when conducting variable sweeps
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for hideandseek-0.1.1.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6b5076d3da46924f115e66677c5b7f829018de84c4cbcacead102b9c5a93b357 |
|
MD5 | f97d0042f6feef9212d5f18de986b42c |
|
BLAKE2b-256 | b58c7c207c41daef3b3fd2f58ed686105aeae971293995f3391f9af778515e6a |