Skip to content

Commit 26997a3

Browse files
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

6 files changed

+640
-588
lines changed

docs/Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ help:
1616

1717
clean:
1818
rm -rf $(BUILDDIR)
19-
rm -rf auto_examples
19+
# rm -rf auto_examples
2020
rm -rf generated
2121
rm -rf gen_modules
2222

0 commit comments

Comments
 (0)