009a1e8404
This new one has a "cheating" top layer, that does not feed down into the unet encoder, but does consume the outputs of the unet. This cheater only operates on half of the input, while the rest of the unet operates on the full input. This limits the dimensionality of this last layer, on the assumption that these last layers consume by far the most computation and memory, but do not require the full input context. Losses are only computed on half of the aggregate input. |
||
|---|---|---|
| .. | ||
| .idea | ||
| data | ||
| models | ||
| scripts | ||
| trainer | ||
| utils | ||
| multi_modal_train.py | ||
| process_video.py | ||
| requirements.txt | ||
| test.py | ||
| train.py | ||
| use_discriminator_as_filter.py | ||