Граф коммитов

2 Коммитов

Автор SHA1 Сообщение Дата
Adam J. Stewart 9489b61372
Remove type ignores for PyTorch (#460)
* Remove type ignores for PyTorch

* Mypy fixes for pytest MonkeyPatch

* Black

* Ignore Identity

* Generic fixes

* Remove unused Generator import

* More fixes

* Fix remaining mypy errors

* More typing cleanups

* typing.OrderedDict isn't available until Python 3.7.2+

* Need Python 3.9 to build docs for fancy OrderedDict

* Fix Python 3.8 and earlier support

* Fix BigEarthNet tests

* Fix bug in ETCI 2021 tests

* Remove unused flake8 ignore

* More robust and well-documented trainer steps

* Many functions don't actually use batch_idx

* Store cast hparams in trainers
2022-03-14 20:35:37 +00:00
estherrolf 0d4811b752
functionality for learning on the prior with QR loss and ChesapeakeCVPR data (#202)
* adding QR loss functions for learning on the prior

* chesapake learn on prior trainer with self-contained code for visualization

* adding prior dataset to the chesapeake datasets; doesn't handle downloading or anything like that yet

* updating init files to include chesapeake CVPR prior

* adding FCNModified for learning on the prior

* changing input to samplers to pass dataset instead of dataset.index

* fixing style issues

* Removing FCN_modified

* Fixing super call and mypy in FCN model

* Added learning on the prior extension

* Update tests

* Formatting

* Adding QR loss

* Added losses to docs

* Removing trainer, moving datamodule

* Combining chesapeake and chesapeake prior datamodules

* Formatting

* Test coverage

* Formatting

* Adding losses

* Re-moving the datamodules around

* Make loss function a torch Module

* Version added

* Fixed some stuff that got messed up in the rebase

* Formatting

* How'd this get there?

* Change qr losses to expect probabilities instead of log-probabilities

* Clean up test

* Rename qr loss file

* Renamed test file

Co-authored-by: Caleb Robinson <calebrob6@gmail.com>
2021-12-28 14:56:14 -06:00