Skip to main content
. 2021 Mar 16;29(7):10000–10035. doi: 10.1364/OE.418296

Table 1. Performance data of all test cases shown in Sec. 3., using the compute platforms described in Sec. 3.1. Walltimes shown in seconds do not include the time spent for saving intermediate results, or for providing diagnostic checkpoint results after each epoch.

Case Backend Platform Tile size Number of tiles per batch Mean walltime per batch (s) Data size (GB)
MDH (simulation) PyTorch Godzilla 512×512 4 (1 set of 4 distances) 0.08 0.004
MDH (experimental) PyTorch Godzilla 2160×2560 3 (1 set of 3 distances) 1.4 166 (all angles)
2D ptychography w/ pos. refin. PyTorch Godzilla 256×256 (5 modes) 35 0.4 2.6
Sparse MS ptychography PyTorch Godzilla 128×128 101 0.3 0.19
Ptychotomography PyTorch Godzilla 64×64 (5 modes) 50 5.2/2.2 (w/wo. tilt refin.) 29.5
Tomography with DO Autograd Theta 25×52 1 5.8 4.8