update

4 months ago · d712005769
parent 75339784c5
commit d712005769
5 changed files with 1081 additions and 229 deletions
--- a/2
+++ b/2
@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-runtime
+FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime

 ARG USER
 ARG GROUP
--- a/Propagator_QuickDraw.ipynb
+++ b/Propagator_QuickDraw.ipynb
--- a/build.sh
+++ b/build.sh
--- a/start.sh
+++ b/start.sh
--- a/utils.py
+++ b/utils.py
@ -15,6 +15,7 @@ def imshow(tensor, figsize=None, title="", **args):
            plt.suptitle("{} {}".format(idx, title))
        return

+    print(type(tensor))
    tensor = tensor.detach().cpu() if type(tensor) == torch.Tensor else tensor
    if tensor.dtype == torch.complex64:
        f, ax = plt.subplots(1, 5, figsize=figsize, gridspec_kw={'width_ratios': [46.5,3,1,46.5,3]})