equivariant_imaging.py π§ΒΆ
Source: scripts/example_scripts/equivariant_imaging.py
Warning
This file has not yet received a complete narrative and docstring audit. Its public source-level API is listed automatically below.
- device
- savefolder
- final_result_fname
- checkpoint_fname
- validation_fname
- experiment
- lidc_dataset
- lidc_dataset
- batch_size
- lidc_dataloader
- lidc_test
- lidc_val
- model
- train_param
- loss_fn
- optimiser
- solver
- my_ssim(x: torch.Tensor, y: torch.Tensor)
No docstring is available.
- class TrainParams
No docstring is available.