Skip to content

API Reference

API Reference

EvoLite

  • __init__(model, dataset): Initialize EvoLite with a model and dataset.
  • compress(): Compresses the model using EvoLite's evolutionary-based compression techniques.
  • save(path): Saves the compressed model to a file.

Utilities

  • save_model(model, path): Save a PyTorch or TensorFlow model.
  • load_model(model, path): Load a PyTorch or TensorFlow model.
  • validate_dataset(dataset): Validate if the dataset is in the expected format.

Compression Techniques

  • Pruning: Removes unimportant weights and nodes from the model.
  • Quantization: Reduces the precision of the model's weights.
  • etc.