|
a/README.md |
|
b/README.md |
|
... |
|
... |
42 |
|
42 |
|
43 |
```shell |
43 |
```shell |
44 |
python main_torch.py |
44 |
python main_torch.py |
45 |
``` |
45 |
``` |
46 |
|
46 |
|
47 |
> With ecg_model.h5 or ecg_model.pt in the directory, the commands above will load the model and evaluate it. |
47 |
With ecg_model.h5 or ecg_model.pt in the directory, the commands above will load the model and evaluate it. |
48 |
> |
48 |
|
49 |
> If you want to train the model, please remove the model file and run the commands. |
49 |
If you want to train the model, please remove the model file and run the commands. |
50 |
|
50 |
|
51 |
visualize the log in tensorboard: |
51 |
visualize the log in tensorboard: |
52 |
|
52 |
|
53 |
```shell |
53 |
```shell |
54 |
tensorboard --logdir=./logs/<log_dir_name> |
54 |
tensorboard --logdir=./logs/<log_dir_name> |