Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Segformer swi #1292

Closed
wants to merge 23 commits into from
Closed
Changes from 1 commit
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
64cd1b6
Sliding window callback for SegFormer model.
Yael-Baron Jul 18, 2023
6a879c0
Removed callback __init__
Yael-Baron Jul 18, 2023
4b40860
Merge branch 'master' into segformer_SWI
BloodAxe Jul 31, 2023
c79d0f2
Changed the sliding window callback's name and all its dependencies t…
Yael-Baron Jul 31, 2023
80c3f1b
Merge remote-tracking branch 'origin/segformer_SWI' into segformer_SWI
Yael-Baron Jul 31, 2023
1b8a739
A recipe for each variant was created based on a default cityscapes_s…
Yael-Baron Jul 31, 2023
6c9b1a8
Models were uploaded to the model zoo.
Yael-Baron Aug 1, 2023
8ff0b9b
Added unit test for segformer
Yael-Baron Aug 1, 2023
f400bf3
An exception is raised at the beginning of training if model does not…
Yael-Baron Aug 1, 2023
2821ee9
A correction was made in the docstring of SegFormer's __init__
Yael-Baron Aug 1, 2023
4e91c8d
Objects sliding_window_crop_size and sliding_window_stride were conve…
Yael-Baron Aug 1, 2023
1475d13
Refactoring the forward_with_sliding_window logic to an external meth…
Yael-Baron Aug 2, 2023
0ec8a9c
Adding a parameter called transforms_for_sliding_window for the Slidi…
Yael-Baron Aug 2, 2023
8a178fa
Small improvements.
Yael-Baron Aug 3, 2023
58e8696
Merge branch 'master' into segformer_SWI
BloodAxe Aug 4, 2023
ec450ec
Fix for commit f400bf34050ac48567fd5317037a70c3fcca39f8
Yael-Baron Aug 6, 2023
9c7ff0f
Merge remote-tracking branch 'origin/segformer_SWI' into segformer_SWI
Yael-Baron Aug 6, 2023
61ea395
SlidingWindowTest unit test from forward_with_sliding_window_test.py …
Yael-Baron Aug 6, 2023
e2595b5
Fixed check if model supports sliding window inference.
Yael-Baron Aug 6, 2023
e9a6e9f
Added a unit test for sliding window inference.
Yael-Baron Aug 6, 2023
db42e6c
Added integration tests for the segformer's models using sliding wind…
Yael-Baron Aug 6, 2023
4c518de
Small fix
Yael-Baron Aug 6, 2023
fb5b620
Merge branch 'master' into segformer_SWI
BloodAxe Aug 7, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Small fix
Yael-Baron committed Aug 6, 2023
commit 4c518de048356c4e24bb410b67f9325eefadc72d
3 changes: 0 additions & 3 deletions tests/integration_tests/pretrained_models_test.py
Original file line number Diff line number Diff line change
@@ -31,9 +31,6 @@
from super_gradients.training import models
import super_gradients
from super_gradients.training.utils.callbacks.callbacks import SlidingWindowValidationCallback
import torch

torch.cuda.set_device(2)


class PretrainedModelsTest(unittest.TestCase):