Python PretrainedNet Examples

Example #1

class PretrainedAudioNet34(
    PretrainedNet(
        "/mnt/raid5/sebastian/model_checkpoints/audionet_34_pretrain/model.ckpt"
    ),
    AudioNet34,
):
    pass

Example #2

class PretrainedFrequencyNet(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/frequency_ae/version_0/checkpoints/_ckpt_epoch_5.ckpt"
        ),
        FrequencyAE,
):
    pass

Example #3

class PretrainedSyncAudioNet(
    PretrainedNet(
        "/mnt/raid/sebastian/log/runs/TRAIN/sync_audio_net/version_1/checkpoints/_ckpt_epoch_70.ckpt"
    ),
    SyncAudioNet,
):
    pass

Example #4

class PretrainedFFFCResnet152(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/imagenet_ff/version_4/checkpoints/_ckpt_epoch_2.ckpt"
        ),
        ImageNetResnet152,
):
    pass

Example #5

class PretrainedFFSyncNetGeneralize(
    PretrainedNet(
        "/home/sebastian/log/showcasings/18_ff_syncnet/ff_sync_net_4_classes_resnet18_100_samples/pre-training/_ckpt_epoch_34.ckpt"
    ),
    FFSyncNet,
):
    pass

Example #6

class R2Plus1SmallAudiolikePretrained(
        PretrainedNet(
            "/mnt/raid/sebastian/model_checkpoints/ff/r2plus1_with_mlp/model_epoch_4.ckpt"
        ),
        R2Plus1SmallAudiolike,
):
    pass

Example #7

class PretrainedBiggerWindowedAECorrectLossStrided(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/windowed_fourier_ae/version_3/checkpoints/_ckpt_epoch_5.ckpt"
        ),
        BiggerWindowedFourierAE,
):
    pass

Example #8

class PretrainedBiggerFourierCorrectLoss(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/bigger_fourier_ae/version_3/checkpoints/_ckpt_epoch_2.ckpt"
        ),
        BiggerFourierAE,
):
    pass

Example #9

class PretrainedBiggerL1AE(
        PretrainedNet(
            "/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/fourier/l1.ckpt"
        ),
        BiggerL1AE,
):
    pass

Example #10

class PretrainedLaplacianLossNet(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/laplacian_loss/version_2/checkpoints/_ckpt_epoch_4.ckpt"
        ),
        LaplacianLossNet,
):
    pass

Example #11

class PretrainedBiggerFourierAE(
        PretrainedNet(
            "/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/fourier/fourier_faulty_loss.ckpt"
        ),
        BiggerFourierAE,
):
    pass

Example #12

class SimpleAEVggPretrained(
        PretrainedNet(
            "/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/vgg/model_ported.ckpt"
        ),
        SimpleAEVGG,
):
    pass

Example #13

class SimpleAEL1Pretrained(
        PretrainedNet(
            "/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/l1/model.ckpt"
        ),
        SimpleAEL1,
):
    pass

Example #14

File: calculate_metrics.py Project: fellnerse/forgerydetection

class Five00BatchesResnet18(
    PretrainedNet(
        "/home/sebastian/log/debug/version_32/checkpoints/_ckpt_epoch_1.ckpt"
    ),
    Resnet18,
):
    pass

Example #15

File: ff_sync_net_end2end.py Project: fellnerse/forgerydetection

class FFSyncNetEnd2EndPretrained(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/ff_syncnet_end2end/version_0/checkpoints/_ckpt_epoch_1.ckpt"
        ),
        FFSyncNetEnd2End,
):
    pass

Example #16

class PretrainedBiggerWindowedAELossSummedOverLast4Dims(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/windowed_fourier_ae/version_0/checkpoints/_ckpt_epoch_5.ckpt"
        ),
        BiggerWindowedFourierAE,
):
    pass

Example #17

class PretrainedImageNetResnet(
        PretrainedNet(
            "/mnt/raid5/sebastian/model_checkpoints/imagenet_net/5_epochs_sgd_1.3e-4/model.ckpt"
        ),
        ImageNetResnet,
):
    pass

Example #18

File: ae.py Project: fellnerse/forgerydetection

class PretrainedVideoAE(
    PretrainedNet(
        "/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/video/ae/"
        "l1+vgg/8x8x7x7_latent_space.ckpt"
    ),
    VideoAE2,
):
    pass

Example #19

File: ae.py Project: fellnerse/forgerydetection

class PretrainedSmallerVideoAE(
    PretrainedNet(
        "/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/video/ae/"
        "l1+vgg/16x8x14x14_latent_space.ckpt"
    ),
    SmallerVideoAE,
):
    pass

Example #20

File: similarity_stuff.py Project: fellnerse/forgerydetection

class PretrainedSimilarityNet(
    PretrainedNet(
        # "/home/sebastian/log/runs/TRAIN/binary_similarity_net/margin_1/checkpoints/_ckpt_epoch_5.ckpt"
        "/home/sebastian/log/runs/TRAIN/binary_similarity_net/margin_20/checkpoints/_ckpt_epoch_8.ckpt"
    ),
    SimilarityNet,
):
    pass

Example #21

class PretrainedFFSyncNet(
    PretrainedNet(
        "/home/sebastian/log/showcasings/18_ff_syncnet/ff_sync_net_5_classes_resnet18_100_samples/pre-training/_ckpt_epoch_7.ckpt"
        # "/home/sebastian/log/runs/TRAIN/ff_syncnet_relative_bb/version_0/_ckpt_epoch_7.ckpt"
        # "/mnt/raid/sebastian/log/runs/TRAIN/ff_syncnet_cropped_faces_full_FF_only_NT/version_0/checkpoints/_ckpt_epoch_4.ckpt"
    ),
    FFSyncNet,
):
    pass

Example #22

class PretrainingSyncAudioNet(
    PretrainedNet(
        "/mnt/raid/sebastian/log/runs/TRAIN/sync_audio_net/version_1/checkpoints/_ckpt_epoch_23.ckpt"
    ),
    SyncAudioNet,
):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._set_requires_grad_for_module(self.r2plus1, requires_grad=False)

Example #23

class R2Plus1SmallAudioLikePretrain(
        PretrainedNet(
            "/home/sebastian/log/runs/TRAIN/pretrain_r2plus1/version_0/checkpoints/_ckpt_epoch_2.ckpt"
        ),
        R2Plus1SmallAudiolike,
):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._set_requires_grad_for_module(self.r2plus1, requires_grad=False)
        self._set_requires_grad_for_module(self.r2plus1.fc, requires_grad=True)

Example #24

class PretrainedAudioNet(
    PretrainedNet("/data/hdd/model_checkpoints/audionet/13_epochs/model.ckpt"), AudioNet
):
    pass