Hugging Face's logo

zeahub
/
hvae

Model card Files
xet
Community
swpenninga commited on
Commit
4ad45d0
·
verified ·
1 Parent(s): 23cbe68

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -13,7 +13,8 @@ Current architecture on at resolution `256x256x3` (3 video frames per input/outp
13
 
14
  The weights that are currently present can be read as follows:
15
  - `lvh`: trained on `EchonetLVH` dataset
16
- - `ur`: encoder retrained with UniformRandom subsampling (`zea.agent`)
 
17
  - `24`: 24 lines/columns used when subsampling
18
 
19
  📚 A usage tutorial can be found on [Collab](https://colab.research.google.com/github/tue-bmd/zea/blob/main/docs/source/notebooks/models/hvae_model_example.ipynb)
 
13
 
14
  The weights that are currently present can be read as follows:
15
  - `lvh`: trained on `EchonetLVH` dataset
16
+ - `ur`: encoder retrained with UniformRandomLines subsampling [`zea.agent`](https://github.com/tue-bmd/zea/blob/main/zea/agent/selection.py)
17
+ - `ge`: encoder retrained with GreedyEntropy subsampling [`zea.agent`](https://github.com/tue-bmd/zea/blob/main/zea/agent/selection.py)
18
  - `24`: 24 lines/columns used when subsampling
19
 
20
  📚 A usage tutorial can be found on [Collab](https://colab.research.google.com/github/tue-bmd/zea/blob/main/docs/source/notebooks/models/hvae_model_example.ipynb)