simplify example models

bioimage-io · Nov 11, 2024 · 54a908a · 54a908a
1 parent 61cbf32
commit 54a908a
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 8 deletions.
diff --git a/example_descriptions/models/unet2d_diff_output_shape/environment.yaml b/example_descriptions/models/unet2d_diff_output_shape/environment.yaml
diff --git a/example_descriptions/models/unet2d_diff_output_shape/v0_4.bioimageio.yaml b/example_descriptions/models/unet2d_diff_output_shape/v0_4.bioimageio.yaml
@@ -4,7 +4,6 @@ cite:
   - { text: training library, doi: "10.5281/zenodo.5108853" }
   - { text: architecture, doi: "10.1007/978-3-319-24574-4_28" }
 covers: [cover.jpg]
-dependencies: conda:environment.yaml
 description: Different output shape
 documentation: documentation.md
 format_version: 0.4.0
@@ -44,3 +43,4 @@ weights:
     kwargs: { depth: 3, in_channels: 1, initial_features: 16, out_channels: 1 }
     sha256: 2c475874f358eb75dc5f3b6af8d186e4fbf8da3acf43fb6662f0d5de21b0a838
     source: weights.pt
+    pytorch_version: 1.6
diff --git a/example_descriptions/models/unet2d_fixed_shape/environment.yaml b/example_descriptions/models/unet2d_fixed_shape/environment.yaml
diff --git a/example_descriptions/models/unet2d_fixed_shape/v0_4.bioimageio.yaml b/example_descriptions/models/unet2d_fixed_shape/v0_4.bioimageio.yaml
@@ -41,7 +41,6 @@ weights:
   pytorch_state_dict:
     architecture: unet.py:UNet2d
     architecture_sha256: 3aeda0ace2e9a595025fa103a3eae2fd620a95924b3fba00ff5d4451acf4ad53
-    dependencies: conda:environment.yaml
     kwargs:
       {
         depth: 3,
@@ -55,3 +54,4 @@ weights:
       }
     sha256: bcf26f08c26f5ef8f891c08dba2c8812bec7991643fb7c0f7b5d3e7dc321092f
     source: weights.pt
+    pytorch_version: 1.6