cp_examples/moco_pretrain/train_moco.py [18:31]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    TensorToRGB,
)
from covidprognosis.plmodules import XrayDataModule
from torchvision import transforms

from moco_module import MoCoModule


def build_args(arg_defaults=None):
    pl.seed_everything(1234)
    data_config = Path.cwd() / "../../configs/data.yaml"
    tmp = arg_defaults
    arg_defaults = {
        "accelerator": "ddp",
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



cp_examples/sip_finetune/train_sip.py [22:35]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    TensorToRGB,
)
from covidprognosis.plmodules import XrayDataModule
from torchvision import transforms

from sip_finetune import SipModule


def build_args(arg_defaults=None):
    pl.seed_everything(1234)
    data_config = Path.cwd() / "../../configs/data.yaml"
    tmp = arg_defaults
    arg_defaults = {
        "accelerator": "ddp",
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -