Fitting models clean - #195

Draft

arthurmccray wants to merge 55 commits into

devfrom

fitting_models_clean

Commits on Mar 18, 2026

Initial plan
Copilot
committed
9cabc82
initial commit for model based refinment

cophus
authored and
Copilot
committed
7739850
fitting model working!

cophus
authored and
Copilot
committed
6e90b3a
adding docstrings

cophus
authored and
Copilot
committed
f63af0c
fixing CoM behaviour

cophus
authored and
Copilot
committed
24c47b4
updates

cophus
authored and
Copilot
committed
984637c
slight tweaks

cophus
authored and
Copilot
committed
1a4c121
dataset4dstem.from_file consistent with read_4dstem args

arthurmccray
authored and
Copilot
committed
9ff42ca
refactor in progress, base and rendering largely done, working not perfect

arthurmccray
authored and
Copilot
committed
207ba98
adding state saving to ModelDiffraction

arthurmccray
authored and
Copilot
committed
9ba0541
switching to state_dict saving

arthurmccray
authored and
Copilot
committed
fb8aae4
first version of FitBase

arthurmccray
authored and
Copilot
committed
f00f255
cleaning up FitBase and ModelDiffraction

arthurmccray
authored and
Copilot
committed
22227c4
moving more stuff to FitBase

arthurmccray
authored and
Copilot
committed
f91142f
reorganizing classes -- no functional change

arthurmccray
authored and
Copilot
committed
5e08803
splitting off ModelDiffractionVisualizations into separate file

arthurmccray
authored and
Copilot
committed
c604a9f
adding hard constraints like force_center for DiskTemplate

arthurmccray
authored and
Copilot
committed
f48e691
adding docstrings and cleaning

arthurmccray
authored and
Copilot
committed
fad203b
adding visualizations and overlays

arthurmccray
authored and
Copilot
committed
fd34ae5
adding turning on/off individual components and parameters

arthurmccray
authored and
Copilot
committed
38cb7bd
fixing center disk duplication and a couple viz bugs

arthurmccray
authored and
Copilot
committed
fb4315d
adding back parameter bounds

arthurmccray
authored and
Copilot
committed
cf7f801
updating colormaps

arthurmccray
authored and
Copilot
committed
3f54795
more consistent hard constraints of ranges

arthurmccray
authored and
Copilot
committed
f12f53e
cleaning up naming of Components

arthurmccray
authored and
Copilot
committed
fa8ca66
Fix docstring type hint for file_type parameter in from_file
Copilot
committed
46e83cd
Merge pull request #1 to make create-clean-fittings-branch
arthurmccray
authored
8cc3389
Merge pull request #193 from arthurmccray/fitting_models_clean
arthurmccray
authored
4b15483

Commits on Mar 25, 2026

Merge branch 'dev' into fitting_models_clean
arthurmccray
committed
fbf631c
updating types for OptimizerParams and SchedulerParams
arthurmccray
committed
e7a94ef

Commits on Jun 2, 2026

Merge remote-tracking branch 'upstream/dev' into HEAD
Miti Shah
committed
220fc01

Commits on Jun 5, 2026

strain multi-optimizer and bragg disk detection
Miti Shah
committed
ba74d62
bug fixes for model fititng code transfer
Miti Shah
committed
9153164

Commits on Jun 6, 2026

fixing rotation plot spacing
Miti Shah
committed
5ee6b5e

Commits on Jun 10, 2026

updating imaging-utils to match fitting_models_clean
Miti Shah
committed
78111c4
Revert "updating imaging-utils to match fitting_models_clean"
Miti Shah
committed
9d3b539

Commits on Jun 22, 2026

adding ability to set batch size for cepstral
Mitis User
committed
9ea871c

Commits on Jun 23, 2026

changing data upload for using more gpu
Mitis User
committed
0dc871f

Commits on Jun 24, 2026

adding data preloading for slightly faster computation
mitis1
committed
d26b402

Commits on Jun 25, 2026

changing error function and adding a varible to gaussian background for elliptical distortion
mitis1
committed
7799d83

Commits on Jul 6, 2026

adding gpu save for template generation and setting device for probe calculation
mitis1
committed
c44fe6c

Commits on Jul 9, 2026

allowing for multiple gaussian backgrounds and lattice in batched updates and updating strain plotting
mitis1
committed
0e68ef5

Commits on Jul 12, 2026

allowing for plotting arguments to be set by user
mitis1
committed
882d052

Commits on Jul 19, 2026

Adding L1 loss, fixing mask + lattice
cophus
committed
3e60343

Commits on Jul 20, 2026

Commits on Jul 28, 2026

Commits on Jul 29, 2026

initial rotation implementation
mitis1
committed
d5812bd

Commits on Jul 30, 2026

fixing subpixel dft fit for cepstral, batched mask implementation for model fitting, and optimizer batched implementation
mitis1
committed
5e66014

Commits on Jul 31, 2026

Commits on Aug 9, 2026

initial data consistency implementation
mitis1
committed
c2a9d73

Commits on Aug 20, 2026

Merge pull request #274 from mitis1/strain-multi-optimizer-fitting-models
cophus
authored
e76a6bf