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 | 4 (1 set of 4 distances) | 0.08 | 0.004 | |
MDH (experimental) | PyTorch | Godzilla | 3 (1 set of 3 distances) | 1.4 | 166 (all angles) | |
2D ptychography w/ pos. refin. | PyTorch | Godzilla | (5 modes) | 35 | 0.4 | 2.6 |
Sparse MS ptychography | PyTorch | Godzilla | 101 | 0.3 | 0.19 | |
Ptychotomography | PyTorch | Godzilla | (5 modes) | 50 | 5.2/2.2 (w/wo. tilt refin.) | 29.5 |
Tomography with DO | Autograd | Theta | 1 | 5.8 | 4.8 |