Skip to content

Commit

Permalink
Update README.md (#1358)
Browse files Browse the repository at this point in the history
Changed input required to "Pre trained floating point model (PyTorch/Keras)" instead of "floating point .pt or .keras model"
  • Loading branch information
ServiAmirPM authored Feb 17, 2025
1 parent b151d9d commit 0ec73ad
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ pip install model-compression-toolkit
```
For installing the nightly version or installing from source, refer to the [installation guide](https://github.com/sony/model_optimization/blob/main/INSTALLATION.md).

**Important note**: In order to use MCT, you’ll need to provide a floating point .pt or .keras model as an input.
**Important note**: In order to use MCT, you’ll need to provide a pre-trained floating point model (PyTorch/Keras) as an input.

### Tutorials and Examples

Expand All @@ -60,7 +60,7 @@ QAT (Quantization Aware Training) | High | High (~12-36 GPU hours) | [QAT API](
For each flow, **Quantization core** utilizes various algorithms and hyper-parameters for optimal [hardware-aware](https://github.com/sony/model_optimization/blob/main/model_compression_toolkit/target_platform_capabilities/README.md) quantization results.
For further details, please see [Supported features and algorithms](#high-level-features-and-techniques).

**Required input**: Floating point model - 32bit model in either .pt or .keras format
**Required input**: Pre-trained floating point model (PyTorch/Keras)

**Optional input**: Representative dataset - can be either provided by the user, or generated utilizing the [Data Generation](#data-generation-) capability

Expand Down

0 comments on commit 0ec73ad

Please sign in to comment.