comm_test.yaml

# @package _global_

# to execute this experiment run:
# python run.py experiment=example_simple.yaml

defaults:
  - override /mode: exp.yaml
  - override /trainer: default.yaml
  - override /model: timm_com.yaml
  - override /datamodule: shard_imagenet.yaml
  - override /callbacks: default.yaml
  - override /logger: wandb.yaml
  - override /callbacks: default.yaml

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

# name of the run determines folder name in logs
# it's also accessed by loggers
name: "timm_res"

seed: 12345

trainer:
  # uncomment for very fast testing batches
  # limit_train_batches: 0.1
  # limit_val_batches: 0.1
  min_epochs: 1
  max_epochs: 50
  precision: 16
  accelerator: gpu
  strategy: ddp
  devices:
    # - 1
    # - 2
    - 3

datamodule:
  batch_size: 128

callbacks:
  early_stopping:
    monitor: "teacher/val/acc"
    patience: 10

  model_checkpoint:
    monitor: "teacher/val/acc"

logger:
  wandb:
    log_model: True