コード例 #1
0
ファイル: experiment.py プロジェクト: vianziro/msc-thesis
    ["--width",  "512", "--height",  "512"],
    ["--width", "1024", "--height", "1024"],
    ["--width", "2048", "--height", "2048"]
]

SIMPLEBIG_ARGS = list(product(COMPLEXITIES, DATASIZES, BORDERS))

####################
# Device Arguments #
####################
if system.HOSTNAME == "cec":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"]]
elif system.HOSTNAME == "dhcp-90-060":
    DEVARGS = [["--device-type", "GPU", "--device-count", "1"]]
elif system.HOSTNAME == "florence":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"]]
elif system.HOSTNAME == "monza":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "2"]]
elif system.HOSTNAME == "tim":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "2"],
               ["--device-type", "GPU", "--device-count", "3"],
               ["--device-type", "GPU", "--device-count", "4"]]
elif system.HOSTNAME == "whz5":
    DEVARGS = [["--device-type", "GPU", "--device-count", "1"]]
else:
    io.warn("Unrecognised hostname, no devargs.")
コード例 #2
0
ファイル: experiment.py プロジェクト: ChrisCummins/msc-thesis
DATASIZES = [["--width", "512", "--height", "512"],
             ["--width", "1024", "--height", "1024"],
             ["--width", "2048", "--height", "2048"]]

SIMPLEBIG_ARGS = list(product(COMPLEXITIES, DATASIZES, BORDERS))

####################
# Device Arguments #
####################
if system.HOSTNAME == "cec":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"]]
elif system.HOSTNAME == "dhcp-90-060":
    DEVARGS = [["--device-type", "GPU", "--device-count", "1"]]
elif system.HOSTNAME == "florence":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"]]
elif system.HOSTNAME == "monza":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "2"]]
elif system.HOSTNAME == "tim":
    DEVARGS = [["--device-type", "CPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "1"],
               ["--device-type", "GPU", "--device-count", "2"],
               ["--device-type", "GPU", "--device-count", "3"],
               ["--device-type", "GPU", "--device-count", "4"]]
elif system.HOSTNAME == "whz5":
    DEVARGS = [["--device-type", "GPU", "--device-count", "1"]]
else:
    io.warn("Unrecognised hostname, no devargs.")
コード例 #3
0
ファイル: io_test.py プロジェクト: 50417/DeepFuzzSL
def test_warn():
    out = StringIO()
    io.warn("foo", file=out)
    assert "WARN" == re.search("WARN", out.getvalue()).group(0)
コード例 #4
0
ファイル: test_io.py プロジェクト: ChrisCummins/phd
 def test_warn(self):
     out = StringIO()
     io.warn("foo", file=out)
     self._test("WARN", re.search("WARN", out.getvalue()).group(0))