Commit dc37d5d9 authored by mashun1's avatar mashun1
Browse files

Update ucf101_train.yaml

parent 339bb812
......@@ -25,7 +25,7 @@ learning_rate: 1e-4
ckpt_every: 100
clip_max_norm: 0.1
start_clip_iter: 100000
local_batch_size: 5 # important
local_batch_size: 1 # important
max_train_steps: 1000
global_seed: 3407
num_workers: 8
......@@ -39,4 +39,4 @@ num_classes: 101
use_compile: False
mixed_precision: False
enable_xformers_memory_efficient_attention: False
gradient_checkpointing: False
\ No newline at end of file
gradient_checkpointing: False
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment