You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Running with the separate script: Add Test DataModule and Test TaskModule to support testing models on a separate script.
Manually restricting dataset types and Lightning Module loop types to be executed by running the train.py: Add a parameter --mode to train.py in order to allow for train or validate (and maybe test) options. As the user may want to just run validation for benchmarking with no training using the training configs, Data Modules, and Task Modules, it would be good to add a parameter to allow skipping training and executing validation only (or test only). It may look like that:
if args.mode=="train":
trainer.fit(model, dataloaders=data_module)
elif args.mode=="valid":
trainer.validate(model, dataloaders=data_module)
In this case, we require all train and valid (and maybe) data loaders to be initialized.
The text was updated successfully, but these errors were encountered:
Support two types of testing:
--mode
totrain.py
in order to allow fortrain
orvalidate
(and maybetest
) options. As the user may want to just run validation for benchmarking with no training using the training configs, Data Modules, and Task Modules, it would be good to add a parameter to allow skipping training and executing validation only (or test only). It may look like that:In this case, we require all train and valid (and maybe) data loaders to be initialized.
The text was updated successfully, but these errors were encountered: