Browse Source
Merge pull request #92 from Royalvice/patch-1
Update train.py
main
Jiale Xu
8 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.strategies import DDPStrategy |
|
|
|
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 |
|
|
|
|
|
|
|
|
