Jacky Lee

Rename Normalize and move to nn

14


for #499

for #482. Credits to @tomfinet and @martinloretzzz

- Add inference tests for ResNet50 - skip `test_car` as ResNet50 fails to classify it - Add test case for cannon

Add MLPerf UNet3D model

3

Creating this draft PR for visibility. Changes made: - Add data loader for the KiTS19 dataset - Implement functions to compute DICE score - Implement functions for UNet3D inference Download...

Training UNet3D

2

Changes made: - Implement iterator for training set - Implement transformations - Implement dice cross entropy loss TODO: - Support batch size > 1 - Implement LR warmup - Run...

Get passing Metal tests in CI

20

Changes made: - Add metal tests in GitHub workflow - Fix metal segfaults - Skip tests that fail on GitHub runners (pass on m1) TODO: - Check if skipped tests...