Fix image

.gitignore

@@ -10,3 +10,4 @@ coverage/
 
 __pycache__/
 docs/build
+models/

Dockerfile

@@ -10,6 +10,7 @@ RUN set -ex \
  && apt-get update \
  && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
         git \
+        vim \
         htop \
  && rm -rf /var/cache/apt \
  && rm -rf /var/lib/apt/lists/*
@@ -19,6 +20,8 @@ WORKDIR /code
 COPY ./ /code
 
 RUN conda install pip \
- && pip install --no-cache-dir -r requirements.txt
+ && pip install --no-cache-dir -r requirements.txt \
+ && pip install numpy --upgrade \
+ && conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly
 
 CMD python app.py

cv4e_lecture13/train.py

@@ -9,7 +9,7 @@ import torch.nn as nn
 from torch.optim import Adam
 from tqdm import trange
 
-from . import dataset, model, utils
+from cv4e_lecture13 import dataset, model, utils
 
 log = None
 

cv4e_lecture13/utils.py

@@ -81,7 +81,7 @@ def init_logging():
 def init_seed(seed):
     if seed is not None:
         random.seed(seed)
-        # numpy.random.seed(seed)       # we don't use NumPy in this code, but you would want to set its random number generator seed, too
+        # numpy.random.seed(seed)
         torch.manual_seed(seed)
         torch.cuda.manual_seed(seed)
         cudnn.benchmark = True
@@ -100,7 +100,7 @@ def init_config(config, log):
     if device not in ['cpu']:
         if torch.cuda.is_available():
             cfg['device'] = 'cuda'
-        elif torch.backends.mps.is_available():
+        elif hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
             cfg['device'] = 'mps'
         else:
             log.warning(