摘要: import tensorflow as tf # 设置相关底层配置 physical_devices = tf.config.experimental.list_physical_devices('GPU') assert len(physical_devices) > 0, "Not enoug 阅读全文
posted @ 2020-10-01 19:51 山…隹 阅读(914) 评论(0) 推荐(0)
摘要: import tensorflow as tf class BasicBlock(tf.keras.layers.Layer): def __init__(self, filter_num, stride=1): super(BasicBlock, self).__init__() self.con 阅读全文
posted @ 2020-10-01 18:57 山…隹 阅读(496) 评论(0) 推荐(0)