diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-07-14 22:31:20 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-07-14 22:31:20 +0800 |
| commit | af0b70c013be4baab0bd6b199f0854107aa24513 (patch) | |
| tree | a9efd1bb13ab391687990485baee4dca63956fa7 /simclr/main.py | |
| parent | 3dbe134676402d54659b795a971175cb0921dada (diff) | |
Fix a path problem
Diffstat (limited to 'simclr/main.py')
| -rw-r--r-- | simclr/main.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/simclr/main.py b/simclr/main.py index 0252e3a..0f93716 100644 --- a/simclr/main.py +++ b/simclr/main.py @@ -11,11 +11,10 @@ from torch.utils.data import Dataset from torchvision.datasets import CIFAR10, CIFAR100, ImageNet from torchvision.transforms import transforms -from libs.criteria import InfoNCELoss - path = str(Path(Path(__file__).parent.absolute()).parent.absolute()) sys.path.insert(0, path) +from libs.criteria import InfoNCELoss from libs.datautils import color_distortion, Clip, RandomGaussianBlur, TwinTransform from libs.optimizers import LARS from libs.schedulers import LinearWarmupAndCosineAnneal, LinearLR |