transiteration
commited on
Commit
•
f1765d7
1
Parent(s):
9f80c40
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 |
|