-
Notifications
You must be signed in to change notification settings - Fork 323
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Major revision of Bolts #839
Comments
While adding new features is good. But what do you think of re using them from torchvision? E.g for g / d / c IoU loss are already in torchvision |
If an appropriate feature is available in other projects and is in a stable state (i.e. the package isn't in beta or the feature itself is not beta/unstable/experimental), I'd go for using such feature from the other projects. This might get tricky with new packages as we do not want to have an endless list of dependencies. But, in case of torchvision, we already require it for models, so that is absolutely ok. |
Then in torchvision.ops we already have all 3. Operations and losses. This is from v0.13 latest one. |
Hey @otaj, I am happy to work on the following to start as they are connected:
If that is all good with you? Thanks! 😆 |
Hey @otaj, |
I would like to work on
Follow-up questions on removing optimization policies.
|
Cool stuff! |
Hi @otaj, Would like to work on features around SimCLR
|
Oh my! I go to sleep, and suddenly, it blows up! Thank you, @Ce11an, @BaruchG, @krishnakalyan3, @luca-medeiros, @ArnolFokam! I added all of you to the list ⚡ 🔩 As to whether we should remove LARS and LinearWarmupCosineAnnealingLR - yes, but with a deprecation warning. I will take them on myself. |
I'd love to contribute this awesome project. Cheers |
Hi, @heimish-kyma, awesome! ⚡ I've added you to the list 💪 |
Hello! Can I try helping with |
Awesome, @shivammehta007, thank you! I added you to the list ⚡ |
Hi everyone, but especially those who already signed up for work (@shivammehta007, @heimish-kyma, @ArnolFokam, @luca-medeiros, @krishnakalyan3, @BaruchG, @Ce11an). Please excuse me for "changing the rules" while you're already signed up for work, but this is very much a learning experience for me as well 🧑🏫 Let's all consider #843 a "testing PR" where we can iterate on the process of how will it look like in the end. 🔩 In order to ensure stability and compatibility, we'd like to not raise any warnings in the tests (other than Please note, that it can happen, that these warnings are raised in other features which you haven't signed up for and there are potentially numerous solutions to that:
Thank you! 🔩 ⚡ 💪 |
Can I also take |
@shivammehta007, absolutely! Thank you, you were added to the list! |
Hi, I would like to work on |
Hi @wonbbnote, thank you very much, you were also added to the list! |
Hi @otaj ,
I Plan to improve tests and documentations for these datasets. |
Hi @nishantb06, thanks a lot, added you to the list! |
Hi @otaj, do you mind adding me to the following:
|
@Atharva-Phatak, @matsumotosan, thank you! You were added to the list! ⚡ 🔩 |
@Atharva-Phatak MoCo is very similar and could perhaps reuse some of the code. What do you think of refactoring them both and seeing if some of the code can be shared? |
@Atharva-Phatak let me know if I can help you. Or if you prefer, I can work on MoCo. |
@senarvi I believe if you could take over MoCo, at least for this iteration and in the next iteration, we can fully revise the SSL module as it is not much customizable :) |
Hi, @senarvi, @Atharva-Phatak! I'm sorry for not being too responsive. Taking longer than two weeks is absolutely fine. I honestly don't have a strong preference regarding if you should review SSL module and MoCo together or separately, it seems you are capable of taking that decision into your hands ⚡ |
I am taking these up.
|
I opened a pull request for MoCo and opened some discussion there. Or what's the best place to discuss the details? |
Hi senarvi, since we are focusing on SSL methods mostly I created an issue #929 which highlights things that we need to refactor and we can continue the discussion from there onwards. I have also added few comments to your PR regarding the same. |
Hi, I'd like to work on |
@lijm1358 Assigned :). Thanks :) |
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions. |
I would love to take on some of the RL stuff. Maybe pl_bolts.models.rl.dqn_model.DQN to start? |
@andrewaf1 Assigning that to you ! |
Let's revision Bolts and breathe some fresh air into them! As outlined in #819 and on a Slack channel, we will revisit every single feature within Bolts.
Please sign up for a feature which you'd like to tackle. Once you do so, I will attach your name in the list and you will be expected to open a PR within two weeks. It might be useful to tackle multiple things at once as the feature list is everything top-level from the repository.
Criteria
The criteria for acceptance are simple:
@under_review
from the selected feature 🎉List of features to be reviewed
Features
pl_bolts.callbacks.byol_updates.BYOLMAWeightUpdate
@matsumotosan BYOL weight update callback #867pl_bolts.callbacks.data_monitor.DataMonitorBase
@luca-medeiros Revisioncallbacks.data_monitor
#848pl_bolts.callbacks.data_monitor.ModuleDataMonitor
@luca-medeiros Revisioncallbacks.data_monitor
#848pl_bolts.callbacks.data_monitor.shape2str
@luca-medeiros Revisioncallbacks.data_monitor
#848pl_bolts.callbacks.data_monitor.TrainingDataMonitor
@luca-medeiros Revisioncallbacks.data_monitor
#848pl_bolts.callbacks.knn_online.concat_all_gather
pl_bolts.callbacks.knn_online.KNNOnlineEvaluator
pl_bolts.callbacks.printing.dicts_to_table
pl_bolts.callbacks.printing.PrintTableMetricsCallback
pl_bolts.callbacks.sparseml.SparseMLCallback
pl_bolts.callbacks.ssl_online.set_training
pl_bolts.callbacks.ssl_online.SSLOnlineEvaluator
pl_bolts.callbacks.torch_ort.ORTCallback
pl_bolts.callbacks.variational.LatentDimInterpolator
pl_bolts.callbacks.verification.base.VerificationBase
pl_bolts.callbacks.verification.base.VerificationCallbackBase
pl_bolts.callbacks.verification.batch_gradient.BatchGradientVerification
pl_bolts.callbacks.verification.batch_gradient.BatchGradientVerificationCallback
pl_bolts.callbacks.verification.batch_gradient.collect_tensors
pl_bolts.callbacks.verification.batch_gradient.default_input_mapping
pl_bolts.callbacks.verification.batch_gradient.default_output_mapping
pl_bolts.callbacks.verification.batch_gradient.selective_eval
pl_bolts.callbacks.vision.confused_logit.ConfusedLogitCallback
pl_bolts.callbacks.vision.image_generation.TensorboardGenerativeModelImageSampler
pl_bolts.callbacks.vision.sr_image_logger.SRImageLoggerCallback
pl_bolts.datamodules.async_dataloader.AsynchronousLoader
pl_bolts.datamodules.binary_emnist_datamodule.BinaryEMNISTDataModule
@matsumotosan Binary MNIST/EMNIST Datasets and Datamodules #866pl_bolts.datamodules.binary_mnist_datamodule.BinaryMNISTDataModule
@matsumotosan Binary MNIST/EMNIST Datasets and Datamodules #866pl_bolts.datamodules.cifar10_datamodule.CIFAR10DataModule
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.datamodules.cifar10_datamodule.TinyCIFAR10DataModule
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.datamodules.cityscapes_datamodule.CityscapesDataModule
@lijm1358pl_bolts.datamodules.emnist_datamodule.EMNISTDataModule
@matsumotosan FashionMNIST/EMNIST Datamodules #871pl_bolts.datamodules.experience_source.BaseExperienceSource
pl_bolts.datamodules.experience_source.DiscountedExperienceSource
pl_bolts.datamodules.experience_source.ExperienceSource
pl_bolts.datamodules.experience_source.ExperienceSourceDataset
pl_bolts.datamodules.fashion_mnist_datamodule.FashionMNISTDataModule
@matsumotosan FashionMNIST/EMNIST Datamodules #871pl_bolts.datamodules.imagenet_datamodule.ImagenetDataModule
pl_bolts.datamodules.kitti_datamodule.KittiDataModule
@wonbbnotepl_bolts.datamodules.mnist_datamodule.MNISTDataModule
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.datamodules.sklearn_datamodule.SklearnDataModule
@Ce11anFeature/839 revision sklearn #846pl_bolts.datamodules.sklearn_datamodule.SklearnDataset
@Ce11anFeature/839 revision sklearn #846pl_bolts.datamodules.sklearn_datamodule.TensorDataset
@Ce11anFeature/839 revision sklearn #846Feature -ArrayDataset
#872pl_bolts.datamodules.sr_datamodule.TVTDataModule
pl_bolts.datamodules.ssl_imagenet_datamodule.SSLImagenetDataModule
pl_bolts.datamodules.stl10_datamodule.STL10DataModule
pl_bolts.datamodules.vision_datamodule.VisionDataModule
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.datamodules.vocdetection_datamodule._prepare_voc_instance
pl_bolts.datamodules.vocdetection_datamodule.Compose
pl_bolts.datamodules.vocdetection_datamodule.VOCDetectionDataModule
pl_bolts.datasets.base_dataset.LightDataset
pl_bolts.datasets.cifar10_dataset.CIFAR10
@BaruchGinsert torchvision dependency and write tests for cifar10 #858pl_bolts.datasets.cifar10_dataset.TrialCIFAR10
@BaruchGinsert torchvision dependency and write tests for cifar10 #858pl_bolts.datasets.concat_dataset.ConcatDataset
pl_bolts.datasets.dummy_dataset.DummyDataset
@nishantb06 Added tests, updated doc-strings for Dummy Datasets #865pl_bolts.datasets.dummy_dataset.DummyDetectionDataset
@nishantb06 Added tests, updated doc-strings for Dummy Datasets #865pl_bolts.datasets.dummy_dataset.RandomDataset
@nishantb06 Added tests, updated doc-strings for Dummy Datasets #865pl_bolts.datasets.dummy_dataset.RandomDictDataset
@nishantb06 Added tests, updated doc-strings for Dummy Datasets #865pl_bolts.datasets.dummy_dataset.RandomDictStringDataset
@nishantb06 Added tests, updated doc-strings for Dummy Datasets #865pl_bolts.datasets.emnist_dataset.BinaryEMNIST
@matsumotosan Binary MNIST/EMNIST Datasets and Datamodules #866pl_bolts.datasets.imagenet_dataset._calculate_md5
pl_bolts.datasets.imagenet_dataset._check_integrity
pl_bolts.datasets.imagenet_dataset._check_md5
pl_bolts.datasets.imagenet_dataset._is_gzip
pl_bolts.datasets.imagenet_dataset._is_tar
pl_bolts.datasets.imagenet_dataset._is_targz
pl_bolts.datasets.imagenet_dataset._is_tarxz
pl_bolts.datasets.imagenet_dataset._is_zip
pl_bolts.datasets.imagenet_dataset._verify_archive
pl_bolts.datasets.imagenet_dataset.extract_archive
pl_bolts.datasets.imagenet_dataset.parse_devkit_archive
pl_bolts.datasets.imagenet_dataset.UnlabeledImagenet
pl_bolts.datasets.kitti_dataset.KittiDataset
@lijm1358 Revisiondatasets.kitti_dataset.KittiDataset
#896pl_bolts.datasets.mnist_dataset.BinaryMNIST
@matsumotosan Binary MNIST/EMNIST Datasets and Datamodules #866pl_bolts.datasets.sr_celeba_dataset.SRCelebA
pl_bolts.datasets.sr_dataset_mixin.SRDatasetMixin
pl_bolts.datasets.sr_mnist_dataset.SRMNIST
pl_bolts.datasets.sr_stl10_dataset.SRSTL10
pl_bolts.datasets.ssl_amdim_datasets.CIFAR10Mixed
pl_bolts.datasets.ssl_amdim_datasets.SSLDatasetMixin
pl_bolts.datasets.utils.prepare_sr_datasets
pl_bolts.losses.object_detection.giou_loss
pl_bolts.losses.object_detection.iou_loss
pl_bolts.losses.rl.double_dqn_loss
pl_bolts.losses.rl.dqn_loss
pl_bolts.losses.rl.per_dqn_loss
pl_bolts.losses.self_supervised_learning.AmdimNCELoss
pl_bolts.losses.self_supervised_learning.CPCTask
pl_bolts.losses.self_supervised_learning.FeatureMapContrastiveTask
pl_bolts.losses.self_supervised_learning.nt_xent_loss
pl_bolts.losses.self_supervised_learning.tanh_clip
pl_bolts.metrics.aggregation.accuracy
@BaruchG Metrics #878pl_bolts.metrics.aggregation.mean
@BaruchG Metrics #878pl_bolts.metrics.aggregation.precision_at_k
@BaruchG Metrics #878pl_bolts.metrics.object_detection.giou
@BaruchGpl_bolts.metrics.object_detection.iou
@BaruchGpl_bolts.models.autoencoders.basic_ae.basic_ae_module.AE
@krishnakalyan3pl_bolts.models.autoencoders.basic_ae.basic_ae_module.cli_main
@krishnakalyan3pl_bolts.models.autoencoders.basic_vae.basic_vae_module.cli_main
@krishnakalyan3pl_bolts.models.autoencoders.basic_vae.basic_vae_module.VAE
@krishnakalyan3pl_bolts.models.autoencoders.components.conv1x1
@krishnakalyan3pl_bolts.models.autoencoders.components.conv3x3
@krishnakalyan3pl_bolts.models.autoencoders.components.DecoderBlock
@krishnakalyan3pl_bolts.models.autoencoders.components.DecoderBottleneck
@krishnakalyan3pl_bolts.models.autoencoders.components.EncoderBlock
@krishnakalyan3pl_bolts.models.autoencoders.components.EncoderBottleneck
@krishnakalyan3pl_bolts.models.autoencoders.components.Interpolate
@krishnakalyan3pl_bolts.models.autoencoders.components.resize_conv1x1
@krishnakalyan3pl_bolts.models.autoencoders.components.resize_conv3x3
@krishnakalyan3pl_bolts.models.autoencoders.components.resnet18_decoder
@krishnakalyan3pl_bolts.models.autoencoders.components.resnet18_encoder
@krishnakalyan3pl_bolts.models.autoencoders.components.resnet50_decoder
@krishnakalyan3pl_bolts.models.autoencoders.components.resnet50_encoder
@krishnakalyan3pl_bolts.models.autoencoders.components.ResNetDecoder
@krishnakalyan3pl_bolts.models.autoencoders.components.ResNetEncoder
@krishnakalyan3pl_bolts.models.detection.components.torchvision_backbones._create_backbone_adaptive
pl_bolts.models.detection.components.torchvision_backbones._create_backbone_features
pl_bolts.models.detection.components.torchvision_backbones._create_backbone_generic
pl_bolts.models.detection.components.torchvision_backbones.create_torchvision_backbone
pl_bolts.models.detection.faster_rcnn.backbones.create_fasterrcnn_backbone
pl_bolts.models.detection.faster_rcnn.faster_rcnn_module._evaluate_iou
pl_bolts.models.detection.faster_rcnn.faster_rcnn_module.FasterRCNN
pl_bolts.models.detection.faster_rcnn.faster_rcnn_module.run_cli
pl_bolts.models.detection.retinanet.backbones.create_retinanet_backbone
pl_bolts.models.detection.retinanet.retinanet_module.cli_main
pl_bolts.models.detection.retinanet.retinanet_module.RetinaNet
pl_bolts.models.detection.yolo.yolo_config._create_convolutional
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config._create_layer
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config._create_maxpool
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config._create_route
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config._create_shortcut
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config._create_upsample
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config._create_yolo
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_config.YOLOConfiguration
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers._aligned_iou
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers._corner_coordinates
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.DetectionLayer
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.GIoULoss
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.IoULoss
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.Mish
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.RouteLayer
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.SELoss
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_layers.ShortcutLayer
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_module.Resize
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_module.run_cli
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.detection.yolo.yolo_module.YOLO
@heimish-kyma Revision models.detection.yolo #851pl_bolts.models.gans.basic.basic_gan_module.cli_main
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.models.gans.basic.basic_gan_module.GAN
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.models.gans.basic.components.Discriminator
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.models.gans.basic.components.Generator
@shivammehta007 Reviewing GAN basics, VisionDataModule, MNISTDataModule, CIFAR10DataModule #843pl_bolts.models.gans.dcgan.components.DCGANDiscriminator
@Atharva-Phatak minor dcgan-import fix #921pl_bolts.models.gans.dcgan.components.DCGANGenerator
@Atharva-Phatak minor dcgan-import fix #921pl_bolts.models.gans.dcgan.dcgan_module.cli_main
@Atharva-Phatak minor dcgan-import fix #921pl_bolts.models.gans.dcgan.dcgan_module.DCGAN
@Atharva-Phatak minor dcgan-import fix #921pl_bolts.models.gans.pix2pix.components.DownSampleConv
@BongYang Revision components relatedmodel.gans.pix2pix
#883pl_bolts.models.gans.pix2pix.components.Generator
@BongYang Revision components relatedmodel.gans.pix2pix
#883pl_bolts.models.gans.pix2pix.components.PatchGAN
@BongYang Revision components relatedmodel.gans.pix2pix
#883pl_bolts.models.gans.pix2pix.components.UpSampleConv
@BongYang Revision components relatedmodel.gans.pix2pix
#883pl_bolts.models.gans.pix2pix.pix2pix_module._weights_init
@BongYang Revision components relatedmodel.gans.pix2pix
#883pl_bolts.models.gans.pix2pix.pix2pix_module.Pix2Pix
@BongYang Revision components relatedmodel.gans.pix2pix
#883pl_bolts.models.gans.srgan.components.ResidualBlock
pl_bolts.models.gans.srgan.components.SRGANDiscriminator
pl_bolts.models.gans.srgan.components.SRGANGenerator
pl_bolts.models.gans.srgan.components.VGG19FeatureExtractor
pl_bolts.models.gans.srgan.srgan_module.cli_main
pl_bolts.models.gans.srgan.srgan_module.SRGAN
pl_bolts.models.gans.srgan.srresnet_module.cli_main
pl_bolts.models.gans.srgan.srresnet_module.SRResNet
pl_bolts.models.mnist_module.cli_main
@matsumotosan Revision of MNIST module #873pl_bolts.models.mnist_module.LitMNIST
@matsumotosan Revision of MNIST module #873pl_bolts.models.regression.linear_regression.cli_main
@Ce11anpl_bolts.models.regression.linear_regression.LinearRegression
@Ce11anpl_bolts.models.regression.logistic_regression.cli_main
@Ce11anpl_bolts.models.regression.logistic_regression.LogisticRegression
@Ce11anpl_bolts.models.rl.advantage_actor_critic_model.AdvantageActorCritic
pl_bolts.models.rl.advantage_actor_critic_model.cli_main
pl_bolts.models.rl.common.agents.ActorCriticAgent
@BaruchGpl_bolts.models.rl.common.agents.Agent
@BaruchGpl_bolts.models.rl.common.agents.PolicyAgent
@BaruchGpl_bolts.models.rl.common.agents.SoftActorCriticAgent
@BaruchGpl_bolts.models.rl.common.agents.ValueAgent
@BaruchGpl_bolts.models.rl.common.cli.add_base_args
@BaruchGpl_bolts.models.rl.common.distributions.TanhMultivariateNormal
@BaruchGpl_bolts.models.rl.common.gym_wrappers.BufferWrapper
@BaruchGpl_bolts.models.rl.common.gym_wrappers.DataAugmentation
@BaruchGpl_bolts.models.rl.common.gym_wrappers.FireResetEnv
@BaruchGpl_bolts.models.rl.common.gym_wrappers.ImageToPyTorch
@BaruchGpl_bolts.models.rl.common.gym_wrappers.make_environment
@BaruchGpl_bolts.models.rl.common.gym_wrappers.MaxAndSkipEnv
@BaruchGpl_bolts.models.rl.common.gym_wrappers.ProcessFrame84
@BaruchGpl_bolts.models.rl.common.gym_wrappers.ScaledFloatFrame
@BaruchGpl_bolts.models.rl.common.gym_wrappers.ToTensor
@BaruchGpl_bolts.models.rl.common.memory.Buffer
@BaruchGpl_bolts.models.rl.common.memory.MeanBuffer
@BaruchGpl_bolts.models.rl.common.memory.MultiStepBuffer
@BaruchGpl_bolts.models.rl.common.memory.PERBuffer
@BaruchGpl_bolts.models.rl.common.memory.ReplayBuffer
@BaruchGpl_bolts.models.rl.common.networks.ActorCategorical
@BaruchGpl_bolts.models.rl.common.networks.ActorContinous
@BaruchGpl_bolts.models.rl.common.networks.ActorCriticMLP
@BaruchGpl_bolts.models.rl.common.networks.CNN
@BaruchGpl_bolts.models.rl.common.networks.ContinuousMLP
@BaruchGpl_bolts.models.rl.common.networks.DuelingCNN
@BaruchGpl_bolts.models.rl.common.networks.DuelingMLP
@BaruchGpl_bolts.models.rl.common.networks.MLP
@BaruchGpl_bolts.models.rl.common.networks.NoisyCNN
@BaruchGpl_bolts.models.rl.common.networks.NoisyLinear
@BaruchGpl_bolts.models.rl.double_dqn_model.cli_main
pl_bolts.models.rl.double_dqn_model.DoubleDQN
@andrewaf1pl_bolts.models.rl.dqn_model.cli_main
pl_bolts.models.rl.dqn_model.DQN
pl_bolts.models.rl.dueling_dqn_model.cli_main
pl_bolts.models.rl.dueling_dqn_model.DuelingDQN
pl_bolts.models.rl.noisy_dqn_model.cli_main
pl_bolts.models.rl.noisy_dqn_model.NoisyDQN
pl_bolts.models.rl.per_dqn_model.cli_main
pl_bolts.models.rl.per_dqn_model.PERDQN
pl_bolts.models.rl.ppo_model.cli_main
pl_bolts.models.rl.ppo_model.PPO
pl_bolts.models.rl.reinforce_model.cli_main
pl_bolts.models.rl.reinforce_model.Reinforce
pl_bolts.models.rl.sac_model.cli_main
pl_bolts.models.rl.sac_model.SAC
pl_bolts.models.rl.vanilla_policy_gradient_model.cli_main
pl_bolts.models.rl.vanilla_policy_gradient_model.VanillaPolicyGradient
pl_bolts.models.self_supervised.amdim.amdim_module.AMDIM
pl_bolts.models.self_supervised.amdim.amdim_module.cli_main
pl_bolts.models.self_supervised.amdim.amdim_module.generate_power_seq
pl_bolts.models.self_supervised.amdim.datasets.AMDIMPatchesPretraining
pl_bolts.models.self_supervised.amdim.datasets.AMDIMPretraining
pl_bolts.models.self_supervised.amdim.networks.AMDIMEncoder
pl_bolts.models.self_supervised.amdim.networks.Conv3x3
pl_bolts.models.self_supervised.amdim.networks.ConvResBlock
pl_bolts.models.self_supervised.amdim.networks.ConvResNxN
pl_bolts.models.self_supervised.amdim.networks.FakeRKHSConvNet
pl_bolts.models.self_supervised.amdim.networks.MaybeBatchNorm2d
pl_bolts.models.self_supervised.amdim.networks.NopNet
pl_bolts.models.self_supervised.amdim.transforms.AMDIMEvalTransformsCIFAR10
pl_bolts.models.self_supervised.amdim.transforms.AMDIMEvalTransformsImageNet128
pl_bolts.models.self_supervised.amdim.transforms.AMDIMEvalTransformsSTL10
pl_bolts.models.self_supervised.amdim.transforms.AMDIMTrainTransformsCIFAR10
pl_bolts.models.self_supervised.amdim.transforms.AMDIMTrainTransformsImageNet128
pl_bolts.models.self_supervised.amdim.transforms.AMDIMTrainTransformsSTL10
pl_bolts.models.self_supervised.byol.byol_module.BYOL
@matsumotosan Revision of BYOL module and tests #874pl_bolts.models.self_supervised.byol.byol_module.cli_main
@matsumotosan Revision of BYOL module and tests #874pl_bolts.models.self_supervised.byol.models.MLP
@matsumotosan Revision of BYOL module and tests #874pl_bolts.models.self_supervised.byol.models.SiameseArm
@matsumotosan Revision of BYOL module and tests #874pl_bolts.models.self_supervised.cpc.cpc_finetuner.cli_main
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.cpc_module.cli_main
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.cpc_module.CPC_v2
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.networks.conv1x1
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.networks.conv3x3
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.networks.cpc_resnet101
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.networks.cpc_resnet50
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.networks.CPCResNet
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.networks.LNBottleneck
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.transforms.CPCEvalTransformsCIFAR10
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.transforms.CPCEvalTransformsImageNet128
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.transforms.CPCEvalTransformsSTL10
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.transforms.CPCTrainTransformsCIFAR10
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.transforms.CPCTrainTransformsImageNet128
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.cpc.transforms.CPCTrainTransformsSTL10
@matsumotosan Revision of CPC module and tests #902pl_bolts.models.self_supervised.evaluator.Flatten
pl_bolts.models.self_supervised.evaluator.SSLEvaluator
pl_bolts.models.self_supervised.moco.callbacks.MocoLRScheduler
pl_bolts.models.self_supervised.moco.moco2_module.cli_main
pl_bolts.models.self_supervised.moco.moco2_module.concat_all_gather
pl_bolts.models.self_supervised.moco.moco2_module.Moco_v2
pl_bolts.models.self_supervised.moco.transforms.GaussianBlur
pl_bolts.models.self_supervised.moco.transforms.Moco2EvalCIFAR10Transforms
pl_bolts.models.self_supervised.moco.transforms.Moco2EvalImagenetTransforms
pl_bolts.models.self_supervised.moco.transforms.Moco2EvalSTL10Transforms
pl_bolts.models.self_supervised.moco.transforms.Moco2TrainCIFAR10Transforms
pl_bolts.models.self_supervised.moco.transforms.Moco2TrainImagenetTransforms
pl_bolts.models.self_supervised.moco.transforms.Moco2TrainSTL10Transforms
pl_bolts.models.self_supervised.resnets._resnet
@luca-medeirospl_bolts.models.self_supervised.resnets.BasicBlock
@luca-medeirospl_bolts.models.self_supervised.resnets.Bottleneck
@luca-medeirospl_bolts.models.self_supervised.resnets.conv1x1
@luca-medeirospl_bolts.models.self_supervised.resnets.conv3x3
@luca-medeirospl_bolts.models.self_supervised.resnets.ResNet
@luca-medeirospl_bolts.models.self_supervised.resnets.resnet101
@luca-medeirospl_bolts.models.self_supervised.resnets.resnet152
@luca-medeirospl_bolts.models.self_supervised.resnets.resnet18
@luca-medeirospl_bolts.models.self_supervised.resnets.resnet34
@luca-medeirospl_bolts.models.self_supervised.resnets.resnet50
@luca-medeirospl_bolts.models.self_supervised.resnets.resnext101_32x8d
@luca-medeirospl_bolts.models.self_supervised.resnets.resnext50_32x4d
@luca-medeirospl_bolts.models.self_supervised.resnets.wide_resnet101_2
@luca-medeirospl_bolts.models.self_supervised.resnets.wide_resnet50_2
@luca-medeirospl_bolts.models.self_supervised.simclr.simclr_finetuner.cli_main
@ArnolFokampl_bolts.models.self_supervised.simclr.simclr_module.cli_main
@ArnolFokampl_bolts.models.self_supervised.simclr.simclr_module.Projection
@ArnolFokampl_bolts.models.self_supervised.simclr.simclr_module.SimCLR
@ArnolFokampl_bolts.models.self_supervised.simclr.simclr_module.SyncFunction
@ArnolFokampl_bolts.models.self_supervised.simclr.transforms.GaussianBlur
@ArnolFokam Revision of SimCLR transforms #857pl_bolts.models.self_supervised.simclr.transforms.SimCLREvalDataTransform
@ArnolFokam Revision of SimCLR transforms #857pl_bolts.models.self_supervised.simclr.transforms.SimCLRFinetuneTransform
@ArnolFokam Revision of SimCLR transforms #857pl_bolts.models.self_supervised.simclr.transforms.SimCLRTrainDataTransform
@ArnolFokam Revision of SimCLR transforms #857pl_bolts.models.self_supervised.simsiam.models.MLP
@matsumotosan Revision of SimSiam module and tests #891pl_bolts.models.self_supervised.simsiam.models.SiameseArm
@matsumotosan Revision of SimSiam module and tests #891pl_bolts.models.self_supervised.simsiam.simsiam_module.cli_main
@matsumotosan Revision of SimSiam module and tests #891pl_bolts.models.self_supervised.simsiam.simsiam_module.SimSiam
@matsumotosan Revision of SimSiam module and tests #891pl_bolts.models.self_supervised.ssl_finetuner.SSLFineTuner
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_finetuner.cli_main
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_module.cli_main
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_module.SwAV
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.BasicBlock
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.Bottleneck
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.conv1x1
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.conv3x3
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.MultiPrototypes
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.ResNet
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.resnet18
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.resnet50
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.resnet50w2
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.resnet50w4
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.swav_resnet.resnet50w5
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.transforms.GaussianBlur
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.transforms.SwAVEvalDataTransform
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.transforms.SwAVFinetuneTransform
@Atharva-Phatak swav-improvements #903pl_bolts.models.self_supervised.swav.transforms.SwAVTrainDataTransform
@Atharva-Phatak swav-improvements #903pl_bolts.models.vision.image_gpt.gpt2.Block
@nishantb06pl_bolts.models.vision.image_gpt.gpt2.GPT2
@nishantb06pl_bolts.models.vision.image_gpt.igpt_module._shape_input
@nishantb06pl_bolts.models.vision.image_gpt.igpt_module.cli_main
@nishantb06pl_bolts.models.vision.image_gpt.igpt_module.ImageGPT
@nishantb06pl_bolts.models.vision.pixel_cnn.PixelCNN
pl_bolts.models.vision.segmentation.cli_main
@lijm1358 Revision models.vision.unet, models.vision.segmentation #880pl_bolts.models.vision.segmentation.SemSegment
@lijm1358 Revision models.vision.unet, models.vision.segmentation #880pl_bolts.models.vision.unet.DoubleConv
@lijm1358 Revision models.vision.unet, models.vision.segmentation #880pl_bolts.models.vision.unet.Down
@lijm1358 Revision models.vision.unet, models.vision.segmentation #880pl_bolts.models.vision.unet.UNet
@lijm1358 Revision models.vision.unet, models.vision.segmentation #880pl_bolts.models.vision.unet.Up
@lijm1358 Revision models.vision.unet, models.vision.segmentation #880pl_bolts.optimizers.lars.LARS
@otajpl_bolts.optimizers.lr_scheduler.linear_warmup_decay
@otajpl_bolts.optimizers.lr_scheduler.LinearWarmupCosineAnnealingLR
@otajpl_bolts.transforms.dataset_normalizations.cifar10_normalization
@matsumotosan Revision of dataset normalizations #898pl_bolts.transforms.dataset_normalizations.emnist_normalization
@matsumotosan Revision of dataset normalizations #898pl_bolts.transforms.dataset_normalizations.imagenet_normalization
@matsumotosan Revision of dataset normalizations #898pl_bolts.transforms.dataset_normalizations.stl10_normalization
@matsumotosan Revision of dataset normalizations #898pl_bolts.transforms.self_supervised.ssl_transforms.Patchify
pl_bolts.transforms.self_supervised.ssl_transforms.RandomTranslateWithReflect
pl_bolts.utils.arguments.gather_lit_args
pl_bolts.utils.arguments.LightningArgumentParser
pl_bolts.utils.arguments.LitArg
pl_bolts.utils.pretrained_weights.load_pretrained
pl_bolts.utils.self_supervised.torchvision_ssl_encoder
pl_bolts.utils.semi_supervised.balance_classes
pl_bolts.utils.semi_supervised.generate_half_labeled_batches
pl_bolts.utils.semi_supervised.Identity
pl_bolts.utils.shaping.tile
pl_bolts.utils.warnings.warn_missing_pkg
Thank you for your contributions! 💪 🚀 🎉
The text was updated successfully, but these errors were encountered: