class PretrainedAudioNet34(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/audionet_34_pretrain/model.ckpt"
),
AudioNet34,
):
pass
class PretrainedFrequencyNet(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/frequency_ae/version_0/checkpoints/_ckpt_epoch_5.ckpt"
),
FrequencyAE,
):
pass
class PretrainedSyncAudioNet(
PretrainedNet(
"/mnt/raid/sebastian/log/runs/TRAIN/sync_audio_net/version_1/checkpoints/_ckpt_epoch_70.ckpt"
),
SyncAudioNet,
):
pass
class PretrainedFFFCResnet152(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/imagenet_ff/version_4/checkpoints/_ckpt_epoch_2.ckpt"
),
ImageNetResnet152,
):
pass
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
class R2Plus1SmallAudiolikePretrained(
PretrainedNet(
"/mnt/raid/sebastian/model_checkpoints/ff/r2plus1_with_mlp/model_epoch_4.ckpt"
),
R2Plus1SmallAudiolike,
):
pass
class PretrainedBiggerWindowedAECorrectLossStrided(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/windowed_fourier_ae/version_3/checkpoints/_ckpt_epoch_5.ckpt"
),
BiggerWindowedFourierAE,
):
pass
class PretrainedBiggerFourierCorrectLoss(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/bigger_fourier_ae/version_3/checkpoints/_ckpt_epoch_2.ckpt"
),
BiggerFourierAE,
):
pass
class PretrainedBiggerL1AE(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/fourier/l1.ckpt"
),
BiggerL1AE,
):
pass
class PretrainedLaplacianLossNet(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/laplacian_loss/version_2/checkpoints/_ckpt_epoch_4.ckpt"
),
LaplacianLossNet,
):
pass
class PretrainedBiggerFourierAE(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/fourier/fourier_faulty_loss.ckpt"
),
BiggerFourierAE,
):
pass
class SimpleAEVggPretrained(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/vgg/model_ported.ckpt"
),
SimpleAEVGG,
):
pass
class SimpleAEL1Pretrained(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/image/ae/l1/model.ckpt"
),
SimpleAEL1,
):
pass
class Five00BatchesResnet18(
PretrainedNet(
"/home/sebastian/log/debug/version_32/checkpoints/_ckpt_epoch_1.ckpt"
),
Resnet18,
):
pass
class FFSyncNetEnd2EndPretrained(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/ff_syncnet_end2end/version_0/checkpoints/_ckpt_epoch_1.ckpt"
),
FFSyncNetEnd2End,
):
pass
class PretrainedBiggerWindowedAELossSummedOverLast4Dims(
PretrainedNet(
"/home/sebastian/log/runs/TRAIN/windowed_fourier_ae/version_0/checkpoints/_ckpt_epoch_5.ckpt"
),
BiggerWindowedFourierAE,
):
pass
class PretrainedImageNetResnet(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/imagenet_net/5_epochs_sgd_1.3e-4/model.ckpt"
),
ImageNetResnet,
):
pass
class PretrainedVideoAE(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/video/ae/"
"l1+vgg/8x8x7x7_latent_space.ckpt"
),
VideoAE2,
):
pass
class PretrainedSmallerVideoAE(
PretrainedNet(
"/mnt/raid5/sebastian/model_checkpoints/avspeech_ff_100/video/ae/"
"l1+vgg/16x8x14x14_latent_space.ckpt"
),
SmallerVideoAE,
):
pass
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
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
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)
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)
class PretrainedAudioNet(
PretrainedNet("/data/hdd/model_checkpoints/audionet/13_epochs/model.ckpt"), AudioNet
):
pass
