diff --git a/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/tasks/README.md b/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/tasks/README.md index 27358261..72bac9e4 100644 --- a/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/tasks/README.md +++ b/environments/community/pytorch_optimizer_coding/FOB/pytorch_fob/tasks/README.md @@ -9,7 +9,7 @@ We currently have the following tasks: | Name | Dataset | Model | Task | Target Metric | Baseline Score | Baseline Runtime | Hardware | | ------- | ---- | ----- | ---- | ------------- | -------------- | ---------------- | -------- | -| [mnist](mnist) | MNIST | MLP | Image Classification | Top-1 Accuracy | 0.97 | 1 min | 1 gpu | +| [mnist](../../baselines/mnist.yaml) | MNIST | MLP | Image Classification | Top-1 Accuracy | 0.97 | 1 min | 1 gpu | | [classification](classification) | [Imagenet-64x64](https://patrykchrabaszcz.github.io/Imagenet32/) | [Wide ResNet](https://arxiv.org/pdf/1605.07146.pdf) | Image Classification | Top-1 Accuracy | 0.69 | 4h | 4 gpu | | [classification_small](classification_small) | [CIFAR100](https://www.cs.toronto.edu/~kriz/cifar.html) | [Resnet18](https://arxiv.org/pdf/1512.03385.pdf) | Image Classification | Top-1 Accuracy | 0.77 | 10 min | 1 gpu | | [segmentation](segmentation) | [MIT Scene Parse](http://sceneparsing.csail.mit.edu/) | [SegFormer](https://arxiv.org/abs/2105.15203) | Semantic Segmentation | Intersection over Union (IoU) | 0.35 | 5h | 4 gpu |