Update README.md
Browse files
README.md
CHANGED
|
@@ -47,7 +47,7 @@ scripts
|
|
| 47 |
- `predict.py` - TorchPredictor class for making predictions using a PyTorch-based model.
|
| 48 |
- `train.py` - Training process using Ray for distributed training.
|
| 49 |
- `tune.py` - Hyperparameter tuning for Language Model using Ray Tune.
|
| 50 |
-
- `utils.py` - Various utility functions for handling data, setting random seeds, saving and loading dictionaries, etc
|
| 51 |
#### Dataset
|
| 52 |
For training, small portion of the [News Category Dataset](https://www.kaggle.com/datasets/setseries/news-category-dataset) was used, which contains numerous headlines and descriptions of various articles.
|
| 53 |
|
|
|
|
| 47 |
- `predict.py` - TorchPredictor class for making predictions using a PyTorch-based model.
|
| 48 |
- `train.py` - Training process using Ray for distributed training.
|
| 49 |
- `tune.py` - Hyperparameter tuning for Language Model using Ray Tune.
|
| 50 |
+
- `utils.py` - Various utility functions for handling data, setting random seeds, saving and loading dictionaries, etc.
|
| 51 |
#### Dataset
|
| 52 |
For training, small portion of the [News Category Dataset](https://www.kaggle.com/datasets/setseries/news-category-dataset) was used, which contains numerous headlines and descriptions of various articles.
|
| 53 |
|