mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-11-05 14:33:15 +00:00
85 lines
1.7 KiB
YAML
85 lines
1.7 KiB
YAML
TRAIN:
|
|
ENABLE: True
|
|
DATASET: Ssv2
|
|
BATCH_SIZE: 32
|
|
EVAL_PERIOD: 5
|
|
CHECKPOINT_PERIOD: 5
|
|
AUTO_RESUME: True
|
|
CHECKPOINT_EPOCH_RESET: True
|
|
CHECKPOINT_FILE_PATH: /checkpoint/fmetze/neurips_sota/40944587/checkpoints/checkpoint_epoch_00035.pyth
|
|
DATA:
|
|
NUM_FRAMES: 16
|
|
SAMPLING_RATE: 4
|
|
TRAIN_JITTER_SCALES: [256, 320]
|
|
TRAIN_CROP_SIZE: 224
|
|
TEST_CROP_SIZE: 224
|
|
INPUT_CHANNEL_NUM: [3]
|
|
MEAN: [0.5, 0.5, 0.5]
|
|
STD: [0.5, 0.5, 0.5]
|
|
PATH_TO_DATA_DIR: /private/home/mandelapatrick/slowfast/data/ssv2
|
|
PATH_PREFIX: /datasets01/SomethingV2/092720/20bn-something-something-v2-frames
|
|
INV_UNIFORM_SAMPLE: True
|
|
RANDOM_FLIP: False
|
|
REVERSE_INPUT_CHANNEL: True
|
|
USE_RAND_AUGMENT: True
|
|
RE_PROB: 0.0
|
|
USE_REPEATED_AUG: False
|
|
USE_RANDOM_RESIZE_CROPS: False
|
|
COLORJITTER: False
|
|
GRAYSCALE: False
|
|
GAUSSIAN: False
|
|
SOLVER:
|
|
BASE_LR: 1e-4
|
|
LR_POLICY: steps_with_relative_lrs
|
|
LRS: [1, 0.1, 0.01]
|
|
STEPS: [0, 20, 30]
|
|
MAX_EPOCH: 35
|
|
MOMENTUM: 0.9
|
|
WEIGHT_DECAY: 5e-2
|
|
WARMUP_EPOCHS: 0.0
|
|
OPTIMIZING_METHOD: adamw
|
|
USE_MIXED_PRECISION: True
|
|
SMOOTHING: 0.2
|
|
SLOWFAST:
|
|
ALPHA: 8
|
|
VIT:
|
|
PATCH_SIZE: 16
|
|
PATCH_SIZE_TEMP: 2
|
|
CHANNELS: 3
|
|
EMBED_DIM: 768
|
|
DEPTH: 12
|
|
NUM_HEADS: 12
|
|
MLP_RATIO: 4
|
|
QKV_BIAS: True
|
|
VIDEO_INPUT: True
|
|
TEMPORAL_RESOLUTION: 8
|
|
USE_MLP: True
|
|
DROP: 0.0
|
|
POS_DROPOUT: 0.0
|
|
DROP_PATH: 0.2
|
|
IM_PRETRAINED: True
|
|
HEAD_DROPOUT: 0.0
|
|
HEAD_ACT: tanh
|
|
PRETRAINED_WEIGHTS: vit_1k
|
|
ATTN_LAYER: divided
|
|
MODEL:
|
|
NUM_CLASSES: 174
|
|
ARCH: slow
|
|
MODEL_NAME: VisionTransformer
|
|
LOSS_FUNC: cross_entropy
|
|
TEST:
|
|
ENABLE: True
|
|
DATASET: Ssv2
|
|
BATCH_SIZE: 64
|
|
NUM_ENSEMBLE_VIEWS: 1
|
|
NUM_SPATIAL_CROPS: 3
|
|
DATA_LOADER:
|
|
NUM_WORKERS: 4
|
|
PIN_MEMORY: True
|
|
NUM_GPUS: 8
|
|
NUM_SHARDS: 4
|
|
RNG_SEED: 0
|
|
OUTPUT_DIR: .
|
|
TENSORBOARD:
|
|
ENABLE: True
|