Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#211 SG-136: Apply ema only on student (KD)

Merged
Louis Dupont merged 1 commits into Deci-AI:master from deci-ai:feature/SG-136_use_ema_only_on_kd_student
1
2
3
4
5
6
7
8
9
10
  1. batch_size: 8 # batch size for trainset in DatasetInterface
  2. val_batch_size: 8 # batch size for valset in DatasetInterface
  3. dataset_dir: /data/cityscapes # path to imagenet directory (local)
  4. crop_size: [2048, 1024] # crop size (size of net's input)
  5. img_size: 1024
  6. train_loader_drop_last: True # drop the last incomplete batch, if the dataset size is not divisible by the batch size
  7. color_jitter: 0. # Training color jitter param
  8. random_scales: [0.5, 2.] # RandomScale scales param for trainset
  9. eval_scale: 1. # Resacle scale-factor of the validation set
  10. cityscapes_ignored_label: 19 # ignored label idx
Discard
Tip!

Press p or to see the previous file or, n or to see the next file