Skip to content

Instantly share code, notes, and snippets.

View riga's full-sized avatar

Marcel Rieger riga

View GitHub Profile
@riga
riga / custom_rise.css
Last active July 10, 2020 17:57
Custom CSS for RISE presentations
body.rise-enabled div.prompt_container {
background-color: red !important;
}
div#notebook {
background-color: green !important;
}
# coding: utf-8
"""
Command to run the tasks:
luigi --module progress_tasks TaskB --workers 2
"""
import time
import luigi
@riga
riga / slurm_test.py
Last active March 4, 2020 18:07
Law slurm submission test
# coding: utf-8
import time
import law
def test_htcondor():
# load the htcondor contrib package, making all exported objects available under law.htcondor
@riga
riga / testDeepJetPerformance.cc
Last active February 4, 2020 16:45
CMS DeepJet performance test between TF and ONNX
/*
* Performance test of the DeepJet model with both TensorFlow and ONNXRuntime.
* Place this file into a cmssw test directory and add the following to the BuildFile.xml:
*
* <bin name="testTFDeepJetPerformance" file="testRunner.cpp,testDeepJetPerformance.cc">
* <use name="cppunit" />
* <use name="FWCore/Utilities" />
* <use name="PhysicsTools/TensorFlow" />
* <use name="PhysicsTools/ONNXRuntime" />
* </bin>
@riga
riga / cmssw_tf_thread_pools.h
Last active January 28, 2020 13:00
Custom CMSSW TF ThreadPools
// custom TensorFlow ThreadPool implementations for CMSSW
#include "tensorflow/core/lib/core/threadpool.h"
#include "tbb/task_arena.h"
#include "tbb/task_group.h"
// thread pool that schedules all tasks within the caller thread
class NoThreadPoolImpl : public tensorflow::thread::ThreadPoolInterface {
public:
@riga
riga / pyrefs.py
Created January 22, 2020 12:52
Python reference count manipulation
# coding: utf-8
import sys
import gc
import ctypes
# configuration of Py_IncRef and Py_DecRef
_c_inc_ref = ctypes.pythonapi.Py_IncRef
@riga
riga / lbn_usage.py
Last active November 30, 2021 06:52
Minimal LBN example using TF 2 (https://github.com/riga/LBN)
# coding: utf-8
"""
This example demonstrates the usage of the LBN different approaches:
- TF eager mode
- TF AutoGraph
- Keras model / layer
"""
import numpy as np
@riga
riga / test_tf20.sh
Created November 27, 2019 14:23
cmsdist and cmssw setup for tf2.0
action() {
local base="$( /bin/pwd )"
git clone https://github.com/cms-sw/pkgtools.git
git clone --single-branch --branch tf-2.0 https://github.com/riga/cmsdist.git
./pkgtools/cmsBuild -i tf-2.0 -c cmsdist/ --repository cms.week0 --arch slc7_amd64_gcc820 -j 10 build cmssw-tool-conf --no-cleanup
export SCRAM_ARCH="slc7_amd64_gcc820"
export CMSSW_VERSION="CMSSW_11_0_0_pre12"
@riga
riga / patch_graph.py
Created November 27, 2019 13:52
Patch a TensorFlow graph
# coding: utf-8
import os
import tensorflow as tf
def read_constant_graph(graph_path, create_session=True, as_text=False):
graph = tf.Graph()
with graph.as_default():
@riga
riga / read_pdf.py
Last active October 30, 2019 13:20
Read PDF values from MiniAOD
# coding: utf-8
import os
def fwlite_loop(path, handle_data=None, start=0, end=-1, object_type="Event"):
"""
Opens one or more ROOT files defined by *path* and yields the FWLite event. When *handle_data*
is not *None*, it is supposed to be a dictionary ``key -> {"type": ..., "label": ...}``. In that