Skip to content

GitLab

Commit dc37d5d9 authored by mashun1's avatar mashun1
Browse files

Update ucf101_train.yaml

parent 339bb812
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
configs/ucf101/ucf101_train.yaml
View file @ dc37d5d9
...@@ -25,7 +25,7 @@ learning_rate: 1e-4 ...@@ -25,7 +25,7 @@ learning_rate: 1e-4
ckpt_every: 100 ckpt_every: 100
clip_max_norm: 0.1 clip_max_norm: 0.1
start_clip_iter: 100000 start_clip_iter: 100000
local_batch_size: 5 # important local_batch_size: 1 # important
max_train_steps: 1000 max_train_steps: 1000
global_seed: 3407 global_seed: 3407
num_workers: 8 num_workers: 8
...@@ -39,4 +39,4 @@ num_classes: 101 ...@@ -39,4 +39,4 @@ num_classes: 101
use_compile: False use_compile: False
mixed_precision: False mixed_precision: False
enable_xformers_memory_efficient_attention: False enable_xformers_memory_efficient_attention: False
gradient_checkpointing: False gradient_checkpointing: False
\ No newline at end of file
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