Skip to content

Instantly share code, notes, and snippets.

View skeller88's full-sized avatar

Shane Keller skeller88

View GitHub Profile
import json
import time
import warnings
import numpy as np
from tensorflow.keras.callbacks import ModelCheckpoint
class ModelCheckpointGCS(ModelCheckpoint):
"""
# Tensorflow image Dockerfile:
# https://github.com/tensorflow/tensorflow/blob/d73faf5fbb7c8bfbf96cc6334111e4352f209e82/tensorflow/tools/dockerfiles/dockerfiles/gpu-jupyter.Dockerfile
# FROM continuumio/miniconda:4.7.12
# Image Dockerfile:
# https://gitlab.com/nvidia/container-images/cuda/-/blob/ubuntu18.04/10.0/runtime/Dockerfile
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
# https://askubuntu.com/questions/141928/what-is-the-difference-between-bin-sh-and-bin-bash
# bash has more functionality than bin, such as "source"