Zongyuan Yang
7 months ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with
1 additions and
1 deletions
-
train.py
|
|
@ -8,7 +8,7 @@ from pytorch_lightning import seed_everything |
|
|
from pytorch_lightning.trainer import Trainer |
|
|
from pytorch_lightning.trainer import Trainer |
|
|
from pytorch_lightning.strategies import DDPStrategy |
|
|
from pytorch_lightning.strategies import DDPStrategy |
|
|
from pytorch_lightning.callbacks import Callback |
|
|
from pytorch_lightning.callbacks import Callback |
|
|
from pytorch_lightning.utilities import rank_zero_only |
|
|
from pytorch_lightning.utilities import rank_zero_only, rank_zero_warn |
|
|
|
|
|
|
|
|
from src.utils.train_util import instantiate_from_config |
|
|
from src.utils.train_util import instantiate_from_config |
|
|
|
|
|
|
|
|
|
