Update README.md
Browse files
README.md
CHANGED
@@ -11,11 +11,11 @@ Accuracy: 95.6%
|
|
11 |
|
12 |
Key Training Params:
|
13 |
|
14 |
-
Steps: 100
|
15 |
-
Batch Size: 64
|
16 |
-
Optimizer: Adam_8bit
|
17 |
-
lr_schedular: cosine_with_restarts
|
18 |
-
lr_scheduler_kwargs = { "num_cycles": len(trainer.get_train_dataloader()) / 100 }
|
19 |
|
20 |
The datasets are an 80/20 partition of the TON-Iot Network dataset.
|
21 |
|
|
|
11 |
|
12 |
Key Training Params:
|
13 |
|
14 |
+
* Steps: 100
|
15 |
+
* Batch Size: 64
|
16 |
+
* Optimizer: Adam_8bit
|
17 |
+
* lr_schedular: cosine_with_restarts
|
18 |
+
* lr_scheduler_kwargs = { "num_cycles": len(trainer.get_train_dataloader()) / 100 }
|
19 |
|
20 |
The datasets are an 80/20 partition of the TON-Iot Network dataset.
|
21 |
|