An open API service indexing awesome lists of open source software.

https://github.com/khaykingleb/research-playground

Efficient ML/DL implementations across multiple domains with K3s multi-node training setup
https://github.com/khaykingleb/research-playground

automatic-speaker-recognition hydra k3s k8s key-word-spotting opentofu pytorch-lightning terraform text-to-speech

Last synced: about 2 months ago
JSON representation

Efficient ML/DL implementations across multiple domains with K3s multi-node training setup

Awesome Lists containing this project

README

        

# Research Playground
> Inspired by [ashleve/lightning-hydra-template](https://github.com/ashleve/lightning-hydra-template) and [NVIDIA/NeMo](https://github.com/NVIDIA/NeMo)

It's a library I created for efficient ML/DL research on various tasks.

## Key features:
- πŸš€ Production-ready training pipelines
- 🧠 Actual model implementations
- ⚑️ Easy configuration management with Hydra
- πŸ“Š Experiment tracking with Weights & Biases
- πŸ”§ Modular architecture for quick prototyping
- 🐳 Docker support for reproducible environments
- ☸️ Multi-GPU training with K3s and Terraform (soon)

## Getting Started

1. Install [asdf](https://asdf-vm.com/guide/getting-started.html) to manage different tools' runtime versions.

2. Update `.env.example` to your needs.

3. Setup your training Hydra config in `configs/experiments/` folder.

4. Choose between local development outside or inside Docker container.

* Outside of Docker (not recommended):

```bash
make init-local
poetry shell && python3 src train --experiment
```
* Inside Docker:

```bash
make init && make build && make run
python3 src train --experiment
```

## Notes

* Use `make help` to see all available commands.

* Use `python3 src --help` to see all available CLI arguments.