Update train_unconditional.py (#3899)

Commit

2 years ago

Update train_unconditional.py (#3899) increase the time of timeout when using big dataset or high resolution