Add MLPerf UNet3D model (#775)

* Add ResNet inference test and cannon

* Test with ResNet50

* test_car works with resnet fix

* Add KiTS19 dataset

* KiTS19: Implement iterate

* No batch load for this dataset

* Save results on iterate

* Implement dice score

* Add data prep and eval functions

* Resolve shape issue

* Conversion works but wrong values

* Segfaults when load_from_pretrained is called

* Fix segfault and assign properly

* Final result generated, though very slow

* Store and load final result to save time

* Fix typo in finalize

* Score computes

* More bug fixes, dice score is very low

* Working broken code

* Assign output values to result

* Getting a much higher score now

* Fix dataset preprocessing

* Mean DICE score of 88.5

* Ugh, typo

* Attempt to reimplement model

* Rename layers

* Tiny model works, kinda

* Accuracy? gone

* Implement InstanceNorm and match torch

* Test instance norm 2d and 3d

* Combined input block with downsample block

* Tiny model works, support strided convtranspose

* Commands to download dataset

* Clean up a bit

* unet3d_v2 -> unet3d

* Remove duplicated code

* Oops, put tests back
This commit is contained in:
Jacky Lee
2023-05-28 20:38:19 -07:00
committed by GitHub
parent 65d09031f2
commit 5d212864b5
9 changed files with 297 additions and 31 deletions

1
.gitignore vendored
View File

@@ -23,6 +23,7 @@ disassemblers/cuda_ioctl_sniffer
datasets/cifar-10-python.tar.gz
datasets/librispeech/
datasets/imagenet/
datasets/kits19/
datasets/squad/
datasets/img_align_celeba*
datasets/open-images-v6-mlperf