docs: update README and library documentation for new compat and CNA modules

- Added `compat` module details in README, including activation and normalization wrappers.
- Documented `CNA2d` block (Conv2d + Normalization + Activation) in both README and library.
- Minor adjustment to environment variables in `.run/Docs.run.xml`.

Author: crutcher

.run/Docs.run.xml

@@ -5,6 +5,7 @@
     <option name="workingDirectory" value="file://$PROJECT_DIR$" />
     <envs>
       <env name="CINIC10_PATH" value="/media/Data/CINIC-10" />
+      <env name="DISPLAY" value=":1" />
     </envs>
     <option name="emulateTerminal" value="true" />
     <option name="channel" value="STABLE" />

crates/bimm/README.md

@@ -36,15 +36,18 @@ let model: ResNet<Wgpu> = prefab
 ```
 
 * [bimm::cache](https://docs.rs/bimm/latest/bimm/cache) - weight loading cache.
+* [bimm::compat](https://docs.rs/bimm/latest/bimm/compat) - future-porting burn mechanisms.
+    * [bimm::compat::activation_wrapper::Activation](https://docs.rs/bimm/latest/bimm/compat/activation_wrapper/enum.Activation.html) -
+      activation layer abstraction wrapper.
+    * [bimm::compat::normalization_wrapper::Normalization](https://docs.rs/bimm/latest/bimm/compat/normalization_wrapper/enum.Normalization.html) -
+      normalization layer abstraction wrapper.
 * [bimm::layers](https://docs.rs/bimm/latest/bimm/layers) - reusable neural network modules.
-    * [bimm::layers::activation](https://docs.rs/bimm/latest/bimm/layers/activation) - activation
-      layers.
-        * [bimm::layers::activation::Activation](https://docs.rs/bimm/latest/bimm/layers/activation/enum.Activation.html)
-            - activation layer abstraction wrapper.
     * [bimm::layers::blocks](https://docs.rs/bimm/latest/bimm/layers/blocks) - miscellaneous
       blocks.
-        * [bimm::layers::blocks::conv_norm](https://docs.rs/bimm/latest/bimm/layers/blocks/conv_norm) -
+        * [bimm::layers::blocks::conv_norm::ConvNorm2d](https://docs.rs/bimm/latest/bimm/layers/blocks/conv_norm) -
           ``Conv2d + BatchNorm2d`` block.
+        * [bimm::layers::blocks::cna::CNA2d](https://docs.rs/bimm/latest/bimm/layers/blocks/conv_norm) -
+          ``Conv2d + Normalization + Activation`` block.
     * [bimm::layers::drop](https://docs.rs/bimm/latest/bimm/layers/drop) - dropout layers.
         * [bimm::layers::drop::drop_block](https://docs.rs/bimm/latest/bimm/layers/drop/drop_block) -
           2d drop

crates/bimm/src/lib.rs

@@ -9,7 +9,8 @@
 //!   * [`compat::normalization_wrapper::Normalization`] - norm layer abstraction wrapper.
 //! * [`layers`] - reusable neural network modules.
 //!   * [`layers::blocks`] - miscellaneous blocks.
-//!     * [`layers::blocks::conv_norm`] - ``Conv2d + BatchNorm2d`` block.
+//!     * [`layers::blocks::cna::CNA2d`] - ``Conv2d + Normalization + Activation`` block.
+//!     * [`layers::blocks::conv_norm::ConvNorm2d`] - ``Conv2d + BatchNorm2d`` block.
 //!   * [`layers::drop`] - dropout layers.
 //!     * [`layers::drop::drop_block`] - 2d drop block / spatial dropout.
 //!     * [`layers::drop::drop_path`] - drop path / stochastic depth.