Problem in log_validation of train_control.py

aigc-apps / CogVideoX-Fun

📹 A more flexible CogVideoX that can generate videos at any resolution and creates videos from images.

Apache License 2.0

443 stars 30 forks source link

Problem in log_validation of train_control.py #53

Open laulampaul opened 1 week ago

laulampaul commented 1 week ago

when I try to run train_control.py , there is a problem in log_validation In README_TRAIN_CONTROL.md , there is no args.train_mode but in the func log_validation

for i in range(len(args.validation_prompts)): with torch.no_grad(): if args.train_mode != "normal":

how can I revise the code and set args.train_mode ?

bubbliiiing commented 1 week ago

Sorry, we haven't fully tested it yet. We need to fix it