Update contrastive learning code

.gitignore

@@ -1 +1,3 @@
-.idea
+.idea
+__pycache__
+deploy.sh

Dockerfile → Dockerfile.python

@@ -23,10 +23,6 @@ RUN mkdir /data
 RUN chmod 775 /data
 RUN chown -R :1337 /data
 
-COPY src /src
-RUN chmod 775 /src
-RUN chown -R :1337 /src
-
 RUN pip3 install transformers
 RUN pip3 install torchcodec
 RUN pip3 install h5py nilearn nibabel
@@ -40,6 +36,10 @@ RUN pip3 install wandb matplotlib scikit-learn
 
 RUN pip3 freeze > pip-freeze.txt
 
+COPY src /src
+RUN chmod 775 /src
+RUN chown -R :1337 /src
+
 WORKDIR /src
 
 ENTRYPOINT ["python3"]

Dockerfile.sweep

@@ -0,0 +1,2 @@
+FROM eidos-service.di.unito.it/barbano/algonauts:latest
+ENTRYPOINT ["/opt/conda/bin/wandb", "agent"]

build.sh

@@ -1,5 +1,8 @@
 #!/bin/sh
 
 # Script to build & deploy your docker image
-docker build -t eidos-service.di.unito.it/barbano/algonauts:latest . -f Dockerfile
-docker push eidos-service.di.unito.it/barbano/algonauts:latest
+docker build -t eidos-service.di.unito.it/barbano/algonauts:latest . -f Dockerfile.python
+docker push eidos-service.di.unito.it/barbano/algonauts:latest
+
+docker build -t eidos-service.di.unito.it/barbano/algonauts:sweep . -f Dockerfile.sweep
+docker push eidos-service.di.unito.it/barbano/algonauts:sweep

requirements.txt

@@ -1,41 +1,90 @@
+absl-py==2.2.2
+annotated-types==0.7.0
+awscli==1.38.34
+botocore==1.37.34
 certifi==2025.1.31
 charset-normalizer==3.4.1
-filelock==3.17.0
-fsspec==2025.3.0
+click==8.1.8
+colorama==0.4.6
+contourpy==1.3.2
+cycler==0.12.1
+docker-pycreds==0.4.0
+docutils==0.19
+filelock==3.18.0
+fonttools==4.57.0
+fsspec==2025.3.2
+gitdb==4.0.12
+GitPython==3.1.44
+grpcio==1.71.0
 h5py==3.13.0
-huggingface-hub==0.29.3
+huggingface-hub==0.30.2
 idna==3.10
+importlib_resources==6.5.2
 Jinja2==3.1.6
+jmespath==1.0.1
 joblib==1.4.2
-lxml==5.3.1
+kiwisolver==1.4.8
+lxml==5.3.2
+Markdown==3.8
 MarkupSafe==3.0.2
+matplotlib==3.10.1
 mpmath==1.3.0
 networkx==3.4.2
 nibabel==5.3.2
 nilearn==0.11.1
-numpy==2.2.3
+numpy==2.2.4
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-cusparselt-cu12==0.6.2
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
 packaging==24.2
 pandas==2.2.3
-pillow==11.1.0
+pillow==11.2.1
+platformdirs==4.3.7
+protobuf==5.29.4
+psutil==7.0.0
+pyasn1==0.6.1
+pydantic==2.11.3
+pydantic_core==2.33.1
+pyparsing==3.2.3
 python-dateutil==2.9.0.post0
-pytz==2025.1
+pytz==2025.2
 PyYAML==6.0.2
 regex==2024.11.6
 requests==2.32.3
+rsa==4.7.2
+s3transfer==0.11.4
 safetensors==0.5.3
 scikit-learn==1.6.1
 scipy==1.15.2
-setuptools==76.0.0
+sentry-sdk==2.26.1
+setproctitle==1.3.5
 six==1.17.0
+smmap==5.0.2
 sympy==1.13.1
-threadpoolctl==3.5.0
-tokenizers==0.21.0
+tensorboard==2.19.0
+tensorboard-data-server==0.7.2
+threadpoolctl==3.6.0
+tokenizers==0.21.1
 torch==2.6.0
-torchaudio==2.6.0
 TorchCodec==0.2.1
 torchvision==0.21.0
 tqdm==4.67.1
-transformers==4.49.0
-typing_extensions==4.12.2
-tzdata==2025.1
-urllib3==2.3.0
+transformers==4.51.3
+triton==3.2.0
+typing-inspection==0.4.0
+typing_extensions==4.13.2
+tzdata==2025.2
+urllib3==2.4.0
+wandb==0.19.9
+Werkzeug==3.1.3
+natsort