arjunraj commited on
Commit
d1316a0
·
verified ·
1 Parent(s): 2d01b23

Publish converted PyTorch checkpoint

Browse files

Converted with Piscis 1.0.0 after exact tensor-level validation of the converter against the published 20230905 JAX/.pt pair.

Files changed (2) hide show
  1. README.md +28 -0
  2. ps_20240419_112256.pt +3 -0
README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: pytorch
3
+ license: mit
4
+ tags:
5
+ - piscis
6
+ - microscopy
7
+ - spot-detection
8
+ - fluorescence-microscopy
9
+ ---
10
+
11
+ # Piscis ps_20240419_112256
12
+
13
+ PyTorch conversion of the legacy Piscis JAX checkpoint
14
+ [`wniu/ps_20240419_112256`](https://huggingface.co/wniu/ps_20240419_112256),
15
+ for Flax-free inference in NimbusImage's Piscis Predict and Train workers.
16
+
17
+ ## Conversion validation
18
+
19
+ The Piscis 1.0.0 JAX-to-PyTorch converter was first validated against the
20
+ published `20230905` JAX and PyTorch checkpoint pair. All 583 tensors matched
21
+ exactly in key, shape, dtype, and value; the maximum absolute difference was
22
+ `0.0`. The same validated converter produced this checkpoint.
23
+
24
+ - Source checkpoint SHA-256: `28fea47454c108dac5e2899f5c4f84fd489792f28fbd5d6b3e002236516952e2`
25
+ - Converted checkpoint SHA-256: `f281a239d1d5075ee2342c4772ba7167c441f878e5345c632436caf3b43d320a`
26
+ - Converted checkpoint size: `30,083,712` bytes
27
+ - State dictionary: 583 tensors
28
+ - Metadata: `adjustment=standardize`, `input_size=(256, 256)`, `dilation_iterations=1`, `channels=1`
ps_20240419_112256.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f281a239d1d5075ee2342c4772ba7167c441f878e5345c632436caf3b43d320a
3
+ size 30083712