DL-Art-School/codes
James Betker bc20b4739e vqvae3
Changes VQVAE as so:
- Reverts back to smaller codebook
- Adds an additional conv layer at the highest resolution for both the encoder & decoder
- Uses LeakyReLU on trunk
2021-01-29 15:24:26 -07:00
..
.idea
data
models vqvae3 2021-01-29 15:24:26 -07:00
scripts
trainer Add SrPixLoss, which focuses pixel-based losses on high-frequency regions 2021-01-25 08:26:14 -07:00
utils
multi_modal_train.py
process_video.py
requirements.txt
test_image_patch_classifier.py
test.py
train.py Add switched_conv with hard routing and make vqvae use it. 2021-01-25 08:25:29 -07:00
train2.py