lucidrains/byol-pytorch
Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
PythonMIT
Issues
- 2
RuntimeError: Only Tensors created explicitly by the user (graph leaves) support the deepcopy protocol at the moment
#52 opened by wengooooo - 0
GPU Memory Usage Extremely High
#94 opened by ClemensSchwarke - 1
[BUG] in trainer.py
#93 opened by ClemensSchwarke - 1
Expected to have finished reduction in the prior iteration before starting a new one.
#92 opened by DingHuilsp - 15
duplicated calculation in gradient
#91 opened by MaryAhn - 3
0.5.6 failing Multiple GPU Error
#58 opened by shivam-aggarwal - 3
About target encoder
#89 opened by ZhaoYuTJPU - 0
Resnet with pre-trained initialization
#88 opened by dpetrini - 0
What does the register_hook do?
#87 opened by YTHmamba - 0
- 5
- 0
The loss has always been around 0.7
#85 opened by liuzuo-byte - 1
Need advice using custom dataset
#60 opened by ramdhan1989 - 0
- 1
Increase EMA-parameter during training
#82 opened by Benjamin-Hansson - 1
How to cluster/predict images?
#78 opened by laxmimerit - 0
loss_fn is not cosine_similarity??
#83 opened by IQ17 - 1
- 2
Why the loss is different from BYOL authors'
#81 opened by Jing-XING - 1
BN layer weights and biases are not updated
#77 opened by kregmi - 3
- 1
For BYOL, cosine similarity of two images should be evaluated by online_encoder or target_encoder?
#75 opened by vincezengqiang - 5
EMA Update of bn buffer
#73 opened by luyvlei - 0
Intutions regarding setting up number of epochs when training on a novel dataset
#71 opened by Sino0904 - 0
- 0
grayscale images
#69 opened by matteoguarrera - 0
Advice of how to avoid cuda out of memory?
#68 opened by HyamsG - 0
- 0
erorr about load ckpt
#66 opened - 4
Can't load ckpt
#38 opened by AndrewTal - 0
How to use the byol train our model?
#65 opened by zhaowanpeng - 1
How to train the better backbone with BYOL?
#49 opened by knaffe - 0
Is there anybody do the experiment to see whether this code base align the accuracy with the BYOL paper?
#64 opened by double-fire-0 - 0
Hi, do you have pretrained checkpoints?
#63 opened by CoinCheung - 5
- 4
Error when use 2 gpus
#53 opened by superwj1990 - 0
error in image augmentation
#61 opened by Jackie-LJQ - 0
Fine-tune VIT model with BYOL method
#59 opened by khawar-islam - 2
- 1
Cannot get embedding
#56 opened by Yurushia1998 - 0
Validation
#55 opened by ivanpanshin - 4
Code error when using torch.nn.DataParallel for multi-gpu: AssertionError: hidden layer avgpool never emitted an output
#50 opened by SilverUnicorn - 1
BYOL Pretrained Model
#51 opened by AhmadM-DL - 2
BYOL collapses
#44 opened by mark-selyaeff - 2
Differences between BYOL and SimSiam
#46 opened by SilverUnicorn - 1
- 2
- 0
example code error with multi-gpu?
#40 opened by sunruina2 - 4
Training loss decreased and then increased
#39 opened by easonyang1996 - 1
CKPT load
#37 opened by ansleepykid