Commit 26997a3
committed
Refactor DTDeepJSCC model to remove MNIST architecture support
- Removed MNIST-specific code from Xie2023DTDeepJSCCEncoder and Xie2023DTDeepJSCCDecoder classes, including initialization parameters and encoder/decoder methods.
- Updated documentation to reflect the removal of MNIST architecture options.
- Simplified encoder and decoder tests by eliminating MNIST-related fixtures and test cases.
- Enhanced the forward methods to ensure compatibility with only CIFAR-10 and custom architectures.
- Improved parallel processing in CapacityAnalyzer by defining a class method for mutual information calculation.
- Added handling for dynamic batch sizes and input sizes in the decoder's forward method.1 parent 086f617 commit 26997a3
File tree
6 files changed
+640
-588
lines changed- docs
- examples/utils
- kaira
- models/image
- utils
- scripts
- tests/models
6 files changed
+640
-588
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
16 | 16 | | |
17 | 17 | | |
18 | 18 | | |
19 | | - | |
| 19 | + | |
20 | 20 | | |
21 | 21 | | |
22 | 22 | | |
| |||
0 commit comments